| .. |
|
__init__.py
|
[LINALG] Add E2E support for `aten.mean.dim` op
|
2022-05-04 20:11:42 +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
|
[LINALG] Add E2E support for `aten.mean.dim` op
|
2022-05-04 20:11:42 +05:30 |
|
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
|
Make e2e testing parallel
|
2022-05-05 21:27:58 -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
|
[LINALG] Add E2E support for `aten.mean.dim` op
|
2022-05-04 20:11:42 +05:30 |
|
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
|
[MLIR][TORCH] E2E support for [ge|ceil].float, [ge|ne|gt].float_int op
|
2022-05-05 21:48:35 +05:30 |
|
scalar_comparison.py
|
[MLIR][TORCH] E2E support for [ge|ceil].float, [ge|ne|gt].float_int op
|
2022-05-05 21:48:35 +05:30 |
|
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 |
|
stats.py
|
[LINALG] Add E2E support for `aten.mean.dim` op
|
2022-05-04 20:11:42 +05:30 |
|
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 |
2ed90741eb