| .. |
|
arange.py
|
[MLIR][TORCH] Add E2E support for torch.arange op
|
2021-12-27 22:45:48 +05:30 |
|
argmax.py
|
add argmax lowering
|
2021-10-13 14:31:16 -04:00 |
|
backprop.py
|
[TORCH][MLIR] Add E2E support for `aten.native_layer_norm_backward` op.
|
2022-02-11 02:03:52 +00:00 |
|
basic.py
|
[LINALG] Fix linalg generic result type argument in TorchToLinalg
|
2022-02-12 03:33:48 +00:00 |
|
constant_alloc.py
|
Fix error in RefineTypes for constant alloc ops (#579)
|
2022-02-10 18:02:18 -08:00 |
|
conv.py
|
Refine static shapes for conv2d and maxpool2d
|
2022-01-03 11:09:23 -06:00 |
|
elementwise.py
|
[TORCH][MLIR] Add E2E support for aten.clone (#571)
|
2022-02-09 19:31:03 -08:00 |
|
histogram_binning_calibration.py
|
Adding an e2e test for histogram binning calibration
|
2022-01-25 18:27:20 -05:00 |
|
index_select.py
|
[MLIR][TORCH] Add E2E support for aten.index_select op
|
2021-12-09 23:13:36 +05:30 |
|
main.py
|
[TORCH][MLIR] Add E2E support for `aten.native_batch_norm` op
|
2022-02-08 02:54:03 +05:30 |
|
matmul.py
|
Add lowering of aten.matmul op.
|
2021-10-26 12:45:09 -04:00 |
|
mlp.py
|
Support aten::linear with rank 3 inputs
|
2021-11-18 22:15:04 +05:30 |
|
nll_loss.py
|
Add aten::nll_loss_backward op
|
2022-02-04 21:57:53 +05:30 |
|
norm_like.py
|
[TORCH][MLIR] Add E2E support for `aten.native_layer_norm_backward` op.
|
2022-02-11 02:03:52 +00:00 |
|
quantized_models.py
|
Dual license the torch-mlir project.
|
2021-10-01 10:46:08 -07:00 |
|
reduction.py
|
Modify softmax decomposition to be more numerically stable.
|
2022-02-03 21:20:36 +05:30 |
|
reshape_like.py
|
[TORCH][MLIR] Add E2E support for `aten._unsafe_view` op.
|
2022-02-10 22:28:58 +05:30 |
|
rng.py
|
[LINALG] Fix linalg generic result type argument in TorchToLinalg
|
2022-02-12 03:33:48 +00:00 |
|
scalar.py
|
add aten.sub.int/aten.mul.int lowering in TorchToStd
|
2021-12-17 10:35:15 -08:00 |
|
slice_like.py
|
Small fixes for slice edge cases (#476)
|
2021-12-15 15:54:41 -06:00 |
|
squeeze.py
|
[TORCH][MLIR] Add E2E support for `aten.squeeze.dim` op
|
2021-12-10 17:01:20 +05:30 |
|
table_batch_embedding.py
|
[TBE] Add a test module for table batch embedding
|
2022-01-28 02:24:28 +05:30 |
|
threshold.py
|
[MLIR][TORCH] Add E2E support for aten.threshold, aten.threshold_backward op
|
2022-01-10 11:56:56 +05:30 |
|
type_conversion.py
|
Convert bool to float or integer type.
|
2022-02-07 21:22:22 +05:30 |
|
type_promotion.py
|
Add scalar type promotion for mul and div (#454)
|
2021-12-03 13:51:25 -06:00 |
|
vision_models.py
|
Add mobilenetv2 and mobilenetv3 to e2e test
|
2022-02-10 19:45:32 -05:00 |
|
xfail_sets.py
|
Add static type information support to `aten.mm`
|
2022-02-15 23:53:42 +00:00 |
fa6cf0bed8