..
__init__.py
Fix version comparison against stable ( #2209 )
2023-06-07 10:19:38 +02:00
arange.py
Revert "[MLIR][TORCH] Fix indentation and spacing for E2E tests"
2022-11-25 12:41:56 +05:30
argmax.py
Revert "build: update llvm tag to 147fe9de"
2022-11-25 12:41:56 +05:30
backprop.py
[MLIR][TORCH] Add support for non-unit stride for conv backward
2023-04-04 17:53:27 +05:30
basic.py
[Torch Dialect] replace none-index in aten.Index.Tensor's param by manually generating it ( #2344 )
2023-08-15 19:36:08 +08:00
cast.py
Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` ( #1890 )
2023-02-21 14:30:05 -08:00
constant_alloc.py
[TOSA] Add support for AtenZerosOp 0/strided layout ( #1983 )
2023-03-30 07:08:20 -07:00
control_flow.py
Revert "build: update llvm tag to 147fe9de"
2022-11-25 12:41:56 +05:30
conv.py
[TOSA] Fix conversion for depthwise convolutions ( #2398 )
2023-08-18 08:15:54 -07:00
custom_op_example.py
Revert "build: update llvm tag to 147fe9de"
2022-11-25 12:41:56 +05:30
elementwise.py
Fix canonicalization of `torch.prim.TupleUnpack`.
2023-07-20 20:08:46 +02:00
elementwise_comparison.py
lib/Conversion/TorchToTosa/TorchToTosa.cpp: Fix legalization of comparions where the input type is bool ( #2304 )
2023-07-17 09:49:04 +02:00
histogram_binning_calibration.py
Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` ( #1890 )
2023-02-21 14:30:05 -08:00
index_select.py
[MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim`
2023-04-14 13:12:56 +05:30
matmul.py
Revert "build: update llvm tag to 147fe9de"
2022-11-25 12:41:56 +05:30
mlp.py
Revert "build: update llvm tag to 147fe9de"
2022-11-25 12:41:56 +05:30
nll_loss.py
[MLIR][TORCH] Add support for the total_weight for aten.nll_loss_forward op
2023-05-30 20:29:27 +05:30
norm_like.py
[Stablehlo] add e2e test for aten.batch_norm ( #2129 )
2023-05-17 09:04:40 -07:00
pooling.py
[Torch Dialect] add support for adaptive_avgpool_1d ( #2342 )
2023-08-05 07:48:09 +08:00
quantized_models.py
Revert "[MLIR][TORCH] Fix indentation and spacing for E2E tests"
2022-11-25 12:41:56 +05:30
reduction.py
[MLIR][TORCH] Add E2E support for cross entropy lowering
2023-04-18 08:00:20 +05:30
reshape_like.py
[TOSA] Add aten.view shape -1 support ( #1815 )
2023-01-20 11:56:26 -08:00
return_types.py
Add float16 support in the refbackend.
2022-12-15 21:19:52 +05:30
rng.py
[Torch Dialect] add folder for aten.add ( #2264 )
2023-06-27 10:55:28 +08:00
scalar.py
Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` ( #1890 )
2023-02-21 14:30:05 -08:00
scalar_comparison.py
Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` ( #1890 )
2023-02-21 14:30:05 -08:00
scatter.py
[MLIR][TORCH] Add support for aten._unsafe_index_put.hacked_twin op
2023-08-11 08:57:01 +05:30
slice_like.py
[Torch-Dialect] emit aten.narrow.Tensor op and decompose it to aten.narrow op ( #2297 )
2023-07-20 16:46:44 +08:00
squeeze.py
[MLIR][TORCH] Add decomposition for prims::squeeze op
2023-04-01 21:45:58 +05:30
stats.py
[MLIR][TORCH] Add E2E support for aten.var_mean.dim op
2023-04-27 22:00:44 +05:30
threshold.py
Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` ( #1890 )
2023-02-21 14:30:05 -08:00
type_conversion.py
[torch-dialect] fix aten.type_as op's folder ( #2283 )
2023-07-20 09:51:58 +08:00
type_promotion.py
Replace `torch.rand` and `torch.randn` in e2e tests with `tu.rand` ( #1890 )
2023-02-21 14:30:05 -08:00
vision_models.py
[cleanup] Remove disabled e2e test
2022-12-07 06:36:48 -08:00
d77b9cf7ae