[E] [TRT] concat: all concat input tensors must have the same dimensions except on the concatenation axis

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.

Thanks

I have the same issue, the model is also defined in NHWC format…