| .. |
|
__init__.py
|
Temporarily revert support for custom op extensions. (#944)
|
2022-06-14 18:24:40 -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
|
Add E2E support for aten.len.str (#969)
|
2022-07-07 10:41:55 -07:00 |
|
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 |
|
custom_op_example.py
|
Allow torch-mlir to support PyTorch extensions. (#895)
|
2022-06-13 14:51:30 -07:00 |
|
elementwise.py
|
linalg: lower `aten.triu` op to `linalg.generic` (#965)
|
2022-06-23 22:45:48 -07:00 |
|
elementwise_comparison.py
|
Add E2E support for AtenAllBoolOp (#874)
|
2022-06-01 18:20:25 -07:00 |
|
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
|
[LINALG] Lower `aten.Matmul` to `linalg.BatchMatmul`
|
2022-06-25 10:58:06 +05:30 |
|
mlp.py
|
Handle `nn.Linear(..., bias=False)` case for TorchToLinalg (#919)
|
2022-06-08 21:13:43 -05: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 decomposition of aten.adaptive_avg_pool2d op
|
2022-05-27 07:56:37 +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
|
[LINALG] Fix handling of size-1 dims in `aten.view` again. (#992)
|
2022-06-30 16:39:25 -07: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 support for bool type in convertScalarToDtype utility
|
2022-06-16 00:00:47 +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 |
f0c3b5a7ed