Is there a way to exclude a layer from layout transformation because the below layer produces an error. I have my original model in NHWC layout and converting to NCHW layout. The model has a series of Conv layers and reshape layers
177 = layout_transform(%176, src_layout=“NCHW”, dst_layout=“NHWC”) an internal invariant was violated while typechecking your program [09:39:55] /home/Documents/tvm/src/tir/ir/data_layout.cc:284: Check failed: src_shape.size() == src_axis.size() (5 vs. 4)