tensorflow/tensorrt

Supporting while loop in tensorflow 2.4

Opened this issue · 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.