torch-mlir/lib/Dialect/Torch
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
..
IR Add aten.ne.bool to make CI pass 2021-10-21 14:45:41 -04:00
Transforms [TORCH][MLIR] Add E2E support for aten.permute. 2021-10-28 10:25:26 -04:00
Utils E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00