mirror of https://github.com/llvm/torch-mlir
afdaa60dd4
The original conversion pattern for `AtenBatchNormOp` required that the input rank be greater than 2; however, the only expectation in the conversion pattern and in Pytorch is that the input rank is greater than 1, since the second dimension of the input must match the size of the `weight`, `bias`, `runningMean`, and `runningVar` inputs. This commit fixes the `inputRank` check. |
||
|---|---|---|
| .. | ||
| test | ||
| torch_mlir | ||
| torch_mlir_e2e_test | ||
| CMakeLists.txt | ||
| TorchMLIRModule.cpp | ||