| .. |
|
__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
|
[MLIR][TORCH] Add E2E support for ScalarImplicit, Int.Scalar op
|
2022-05-10 22:40:49 +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
|
[LINALG] Add support for preserve memory format in aten_empty_like op.
|
2022-05-10 09:37:55 +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
|
[MLIR][TORCH] Add E2E support for aten.any.bool op
|
2022-05-24 17:24: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 support for ceil_mode = true for pooling ops
|
2022-05-11 12:52:47 +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
|
torch,linalg: add support for translating aten.linalg.vector_norm (#839)
|
2022-05-19 15:48:15 -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
|
Add shape info to `rand_like` + support for `dtype` flag (#851)
|
2022-05-12 16:00:59 -07:00 |
|
scalar.py
|
[MLIR][TORCH] Add E2E support for aten.Bool.[float|int] op
|
2022-05-24 21:18:34 +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 |
56e77d4213