Supporting while loop in tensorflow 2.4
Opened this issue · 0 comments
Some-random commented
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.