| .. |
|
argmax.py
|
add argmax lowering
|
2021-10-13 14:31:16 -04:00 |
|
backprop.py
|
Add e2e test for aten.log_softmax_back_data op
|
2021-11-19 00:08:28 +05:30 |
|
basic.py
|
[TORCH][MLIR] Add E2E support for aten._softmax operation. (#431)
|
2021-11-25 11:19:02 +05:30 |
|
batchnorm.py
|
E2e support for layernorm.
|
2021-10-04 14:15:13 -04:00 |
|
conv.py
|
Dual license the torch-mlir project.
|
2021-10-01 10:46:08 -07:00 |
|
elementwise.py
|
Add leakyrelu support
|
2021-11-27 23:04:46 +05:30 |
|
main.py
|
add aten.add.int lowering in TorchToStd
|
2021-11-29 13:22:50 -05:00 |
|
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 |
|
quantized_models.py
|
Dual license the torch-mlir project.
|
2021-10-01 10:46:08 -07:00 |
|
reduction.py
|
Dual license the torch-mlir project.
|
2021-10-01 10:46:08 -07:00 |
|
scalar.py
|
add aten.add.int lowering in TorchToStd
|
2021-11-29 13:22:50 -05:00 |
|
type_conversion.py
|
Add convertScalarToDtype helper.
|
2021-11-08 17:50:52 -05:00 |
|
type_promotion.py
|
Add convertScalarToDtype helper.
|
2021-11-08 17:50:52 -05:00 |
|
view.py
|
[MLIR][TORCH] Add E2E support for `torch.aten.view`
|
2021-10-29 22:33:10 +05:30 |
|
vision_models.py
|
Add min/max/clamp support.
|
2021-10-27 13:29:21 -07:00 |
|
xfail_sets.py
|
lowered addcmul and addcdiv to linalg
|
2021-11-24 17:26:47 -05:00 |
03fdf56f21