I have a saved model which I converted to Onnx using tf2onnx, as follows:
python3 -m tf2onnx.convert --fold_const --saved-model saved_model/ --output model.onnx. The model contains unidirectional LSTM layers. I am then importing this model into tensorRT. The parsing of the model completes properly however I then get the following error when TensorRT tries to validate the network:
(Unnamed Layer* 155) [Recurrence]: recurrence inputs must have same type Loop API is not supported on the requested platform. Network validation failed.
Any advice would be greatly appreciated.
TensorRT 7 :
CUDA 10.2 :
CUDNN 184.108.40.206-1 :
Ubuntu 18.04 64 bit :