llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/e2e_testing/torchscript
History
Gaurav Shukla 442ff4605c [LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm` 
- This commit decomposes the `aten.batch_norm` op into the
  `aten.native_batch_norm` op, instead of lowering it to the
  `linalg.generic` op.
- It also adds run-time asserts in the `aten.native_batch_norm` lowering
  to make sure that the shape of the weight, bias, running_mean, and
  running_var must match the num of features.
- Since the `aten.native_batch_norm` op is not supported at TOSA backend,
  all the modules that are dependent on the `aten.native_batch_norm` op
  will fail and therefore they should be removed from the TOSA `passing`
  set.
- It also moves `checkNotNone` to utility.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 		2022-02-16 23:41:38 +05:30
..
arange.py [MLIR][TORCH] Add E2E support for torch.arange op 2021-12-27 22:45:48 +05:30
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
backprop.py Add e2e test for aten.log_softmax_back_data op 2021-11-19 00:08:28 +05:30
basic.py Modify aten._log_softmax op decomposition for numerical stability. 2022-02-16 12:26:17 +05:30
cast.py [LINALG] Add E2E support for `aten.[Bool.Tensor|Float.Tensor]` op 2022-02-14 23:09:20 +05:30
constant_alloc.py [LINALG] Add value tensor variant to `fill_.Scalar` (#600) 2022-02-15 11:58:03 -08:00
conv.py Refine static shapes for conv2d and maxpool2d 2022-01-03 11:09:23 -06:00
elementwise.py [LINALG] Add E2E support for `aten.[le|ge].Scalar` ops 2022-02-15 12:21:09 +05:30
elementwise_comparison.py [LINALG] Add E2E support for `aten.[le|ge].Scalar` ops 2022-02-15 12:21:09 +05:30
histogram_binning_calibration.py Adding an e2e test for histogram binning calibration 2022-01-25 18:27:20 -05:00
index_select.py [MLIR][TORCH] Add E2E support for aten.index_select op 2021-12-09 23:13:36 +05:30
main.py [LINALG] Add E2E support for `aten.[le|ge].Scalar` ops 2022-02-15 12:21:09 +05:30
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Support aten::linear with rank 3 inputs 2021-11-18 22:15:04 +05:30
nll_loss.py [LINALG] Fix name conflict of `self` keyword. 2022-02-14 22:55:31 +05:30
norm_like.py [LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm` 2022-02-16 23:41:38 +05:30
quantized_models.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
reduction.py Modify softmax decomposition to be more numerically stable. 2022-02-03 21:20:36 +05:30
reshape_like.py [TORCH][MLIR] Add E2E support for `aten._unsafe_view` op. 2022-02-10 22:28:58 +05:30
rng.py [LINALG] Fix linalg generic result type argument in TorchToLinalg (#588) 2022-02-11 19:42:18 -08:00
scalar.py add aten.sub.int/aten.mul.int lowering in TorchToStd 2021-12-17 10:35:15 -08:00
scalar_comparison.py [LINALG] Add E2E support for `aten.eq.int` op 2022-02-15 01:37:35 +05:30
slice_like.py Small fixes for slice edge cases (#476) 2021-12-15 15:54:41 -06:00
squeeze.py [TORCH][MLIR] Add E2E support for `aten.squeeze.dim` op 2021-12-10 17:01:20 +05:30
table_batch_embedding.py [TBE] Add a test module for table batch embedding 2022-01-28 02:24:28 +05:30
threshold.py [tosa] Support for some ops and fix for Issue #532 (#575) 2022-02-11 12:30:02 -08:00
type_conversion.py Convert bool to float or integer type. 2022-02-07 21:22:22 +05:30
type_promotion.py Add scalar type promotion for mul and div (#454) 2021-12-03 13:51:25 -06:00
vision_models.py Add mobilenetv2 and mobilenetv3 to e2e test 2022-02-10 19:45:32 -05:00
xfail_sets.py [LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm` 2022-02-16 23:41:38 +05:30