| .. |
|
__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
|
[LINALG] Add dynamic support for `PrimMinIntOp`
|
2023-06-01 11:14:53 -07: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
|
[MLIR][TORCH] Add support for non-unit stride for conv backward
|
2023-04-04 17:53:27 +05:30 |
|
custom_op_example.py
|
Revert "build: update llvm tag to 147fe9de"
|
2022-11-25 12:41:56 +05:30 |
|
elementwise.py
|
[MLIR][TORCH] Add E2E support for aten.tril op (#2202)
|
2023-06-05 16:17:01 -07:00 |
|
elementwise_comparison.py
|
[Torch Dialect] add support for AtenIsnanOp (#2170)
|
2023-06-07 10:06:27 +08: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
|
Fix default value of `stride` in 2d pooling ops in linalg and tosa (#2065)
|
2023-04-27 08:31:36 -07: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
|
Fix creation of empty tensor in decomposition for randn ops (#2043)
|
2023-04-19 08:25:39 -07: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
|
[TM_TENSOR] Add `aten.scatter.[src|value]` op
|
2023-05-29 12:35:53 +05:30 |
|
slice_like.py
|
[Torch Dialect] Support recompose aten.split.Tensor + prim.ListUnpack (#2192)
|
2023-06-07 01:38:04 +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
|
Revert "build: update llvm tag to 147fe9de"
|
2022-11-25 12:41:56 +05:30 |
|
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 |
816880774b