tensorrt icon indicating copy to clipboard operation
tensorrt copied to clipboard
verified publisher icon tensorflow

Supporting while loop in tensorflow 2.4

Open Some-random opened this issue 4 years ago • 0 comments

I'm wondering whether TensorRT supports while loop under tensorflow 2.4. We want to convert a transformer based model to TensorRT but we're having issue with the while loop in transformer decoder.

Some-random avatar Mar 30 '21 08:03 Some-random