https://github.com/banach-space commented: Thanks for the contribution! My main comment would be that this pass is a bit limited ATM: > Convert conv_2d_nhwc_fhwc to conv_2d_nhwc_hwcf Why not support conversions in both directions? https://github.com/llvm/llvm-project/pull/68567