| .. |
|
__init__.py
|
Add lowering for _convolution
|
2022-07-11 11:03:03 +05:30 |
|
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 lowering to linalg for softplus and log1p
|
2022-07-25 21:25:57 +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 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 lowering for _convolution
|
2022-07-11 11:03:03 +05:30 |
|
custom_op_example.py
|
Allow torch-mlir to support PyTorch extensions. (#895)
|
2022-06-13 14:51:30 -07:00 |
|
elementwise.py
|
Add e2e support for aten.expm1
|
2022-07-27 12:31:35 +05:30 |
|
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
|
Fix typo in `inputRank` check of `AtenBatchNormOp` (#1046)
|
2022-07-15 09:35:59 -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
|
Allow expanding and collapsing in aten::view (#1082)
|
2022-07-20 17:35:51 -04: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
|
[MLIR][TORCH] Add negative dim support for aten.cat and aten.slice op
|
2022-07-18 14:01:33 +05:30 |
|
squeeze.py
|
Move e2e test definitions into the `torch_mlir_e2e_test` package
|
2022-03-25 13:56:41 -07:00 |
|
stats.py
|
[MLIR][TORCH] Add decomposition for aten.var.dim op
|
2022-07-15 09:53: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 |
3c9addf19c