..
__init__.py
[TORCH][MLIR] Fix ConstantPad2dStaticModule test.
2022-04-29 21:57:01 +05:30
arange.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
argmax.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
backprop.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
basic.py
Add aten.flip ( #817 )
2022-05-02 16:01:15 -04:00
cast.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
constant_alloc.py
[MLIR][TORCH] Add E2E support for aten.masked_fill.Scalar op
2022-05-02 22:27:33 +05:30
control_flow.py
[TORCH][MLIR]Add lowering for control flow operations.
2022-04-29 16:25:58 +05:30
conv.py
Add 2D case for convolution ( #693 )
2022-04-08 00:47:57 -04:00
elementwise.py
test: add end-to-end test for aten.neg ( #760 )
2022-04-15 12:37:57 -07:00
elementwise_comparison.py
[LINALG] Add the lowering of `aten.ne.Scalar` op
2022-04-05 21:07:28 +05:30
histogram_binning_calibration.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
index_put.py
Fix the input tensors inplace update issue for e2e tests
2022-04-28 11:43:54 -04:00
index_select.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
matmul.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
mlp.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
nll_loss.py
Add 1D, weight, and reduction support to nll_loss_backward ( #729 )
2022-04-04 10:57:49 -07:00
norm_like.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
pooling.py
[MLIR][TORCH] Add E2E support for aten.avg_pool2d op
2022-05-02 12:31:44 +05:30
quantized_models.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
reduction.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
reshape_like.py
This commit decomposes aten._reshape_alias op into aten.view op. ( #690 )
2022-03-28 23:54:28 -05:00
return_types.py
[REFBACKEND] Add support for returning multiple different return types.
2022-04-21 09:02:30 +05:30
rng.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
scalar.py
Revert "[MLIR][TORCH] Add E2E support for aten.ceil.float op"
2022-04-28 07:49:58 -07:00
scalar_comparison.py
Revert "[MLIR][TORCH] Add E2E support for aten.ge.float op"
2022-04-28 07:49:58 -07:00
slice_like.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
squeeze.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
table_batch_embedding.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
threshold.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
type_conversion.py
[MLIR][TORCH] Add E2E support for aten.to.dtype_layout op
2022-05-03 12:48:58 +05:30
type_promotion.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00
vision_models.py
Move e2e test definitions into the `torch_mlir_e2e_test` package
2022-03-25 13:56:41 -07:00