I use the ‘convert_layout’ pass to convert the calculation format of convolution This is the Relay function after this pass:
And then I write a pass to remove the layout_transform of output. But I Found the output shape is not adjust to the new output shape. How I can solve this problem? @tqchen @junrushao @Hzfengsy
