.. |
__init__.py
|
Fix test fails due to upstream PyTorch change (#793)
|
2022-04-25 12:34:32 -07:00 |
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] Bug fix i64 vs i32 type comparison.
|
2022-04-22 08:09:58 +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 torch.layout information
|
2022-04-07 20:47:49 +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
|
Add support for multi-dim input to `index_put_impl` (#722)
|
2022-03-31 09:27:21 -07: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 max_pool2d_with_indices op
|
2022-04-18 21:05:19 +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
|
Move e2e test definitions into the `torch_mlir_e2e_test` package
|
2022-03-25 13:56:41 -07:00 |
scalar_comparison.py
|
Move e2e test definitions into the `torch_mlir_e2e_test` package
|
2022-03-25 13:56:41 -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
|
Move e2e test definitions into the `torch_mlir_e2e_test` package
|
2022-03-25 13:56:41 -07:00 |
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 |