We have a concat layer in our Tensorflow model. When converting it to TRT, we got the following error:
[E] [TRT] concat: all concat input tensors must have the same dimensions except on the concatenation axis
Here is our code:
inputs = tf.concat([cancat_mv2, prev], axis=3)
Could you please help take a look? It seems that TRT uses the incorrect axis to concat. Our input date format is NHWC so we use “3” as the axis.