TensorRT conv layer output with shape HWC instead of CHW

Hi, I am converting a tensorflow model with data_format channels_first to UFF and parse. The input image is 3, 360, 360 but get output 12, 12, 512. I am confused why the output format is HWC
Thank you!

TensorRT vesion:
System: Ubuntu 18.04
GPU: GeForce 1050Ti
TF version: 1.14.0

And I want to know how the tensorrt insert transpose op when converting tensorflow pb


Could you please share the script & model file to reproduce the issue?


Hi, I have solved the problem, please close it