Facing this error while trying to run a pytorch model

Check failed: (tir::BijectiveLayout(new_src_layout, dst_layout).defined()) is false: Cannot insert layout transform because there are inconvertible layouts: NHW v.s. NCW