torch-mlir/e2e_testing/torchscript
Ramiro Leal-Cavazos afdaa60dd4
Fix typo in `inputRank` check of `AtenBatchNormOp` (#1046)
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.
2022-07-15 09:35:59 -07:00
..
main.py Add -s flag to run e2e tests sequentially 2022-05-11 21:16:41 -04:00
xfail_sets.py Fix typo in `inputRank` check of `AtenBatchNormOp` (#1046) 2022-07-15 09:35:59 -07:00