| .. |
|
__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
|
[MLIR][TORCH] Add StableHLO lowering for embedding_bag.padding_idx op
|
2023-09-05 21:32:23 +05:30 |
|
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
|
[Torch Dialect] add canonicalize for aten.min.other (#2452)
|
2023-09-11 17:28:22 +08: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 `torch.aten.prod_dim_int` (#2423)
|
2023-09-05 13:38:51 -07:00 |
|
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][stablehlo] emit aten.rand op and add converter to stablehlo (#2413)
|
2023-08-27 21:56:36 +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] support aten.split_with_sizes (#2431)
|
2023-09-04 09:59:26 +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 |
1f20b7275d