banach-space wrote: @qedawkins I've not worked much with non-depthwise convs and `inferConvolutionDimsImpl` is quite ... convolved 😅 So, please could you double check this and let me know if I missed sth? https://github.com/llvm/llvm-project/pull/90057