tensorflow/tensorrt

Error converting multilingual universal sentence encoder to TensorRT. Input 1 of node StatefulPartitionedCall was passed float from statefulpartitionedcall_args_1:0 incompatible with expected resource.

sebastinajay opened this issue · 1 comments

I'm getting below error while converting multilingual universal sentence encoder(https://tfhub.dev/google/universal-sentence-encoder-multilingual/3) to TensorRT using TrtGraphConverterV2

InvalidArgumentError: Input 1 of node StatefulPartitionedCall was passed float from statefulpartitionedcall_args_1:0 incompatible with expected resource.

Below is the Colab link : https://colab.research.google.com/drive/1g8jHUMdpMgAQPv_kG3YJhP12SmSdHW-H?usp=sharing

Python Version: 3.6.9
TensorFlow Version: 2.3.0

I'm getting a similar issue.