torch-mlir/lib/Dialect/Torch
Vivek Khandelwal 2597c481f6 [MLIR][TORCH] Add E2E support for aten.new_empty op
This commit decomposes `aten.new_empty` op into `aten.empty.memory_format` op.

This commit also made a dtype fix to the constant tensor allocation like ops.
Earlier the dtype for the result was inferred from the result type; now, it's
being evaluated as per the original definition of the op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-03-30 13:21:01 +05:30
..
IR Add minimal support for Union types. 2022-03-29 17:45:48 -07:00
Transforms [MLIR][TORCH] Add E2E support for aten.new_empty op 2022-03-30 13:21:01 +05:30
Utils [MLIR][TORCH] Add E2E support for aten.new_empty op 2022-03-30 13:21:01 +05:30
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00