torch-mlir/e2e_testing/torchscript
Prateek Gupta c33a2ca952 [TORCH][MLIR] Add E2E support for aten.permute.
This commit adds lowering of aten.permute to linalg.generic operation.

Signed-Off-By: Prateek Gupta <prateek@nod-labs.com>
2021-10-28 10:25:26 -04:00
..
argmax.py add argmax lowering 2021-10-13 14:31:16 -04:00
basic.py [TORCH][MLIR] Add E2E support for aten.permute. 2021-10-28 10:25:26 -04:00
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 min/max/clamp support. 2021-10-27 13:29:21 -07:00
main.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
matmul.py Add lowering of aten.matmul op. 2021-10-26 12:45:09 -04:00
mlp.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00
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
vision_models.py Add min/max/clamp support. 2021-10-27 13:29:21 -07:00
xfail_sets.py Add min/max/clamp support. 2021-10-27 13:29:21 -07:00