torch-mlir/lib/Dialect/Torch
Vivek Khandelwal ddd45d6068 [MLIR][TORCH] Add E2E support for aten.new_zeros, aten.new_ones op
This commit adds lowering of `aten.new_zeros` and `aten.new_ones` op

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2022-03-01 22:09:47 +05:30
..
IR LLVM bump 2022-02-16 15:28:13 -05:00
Transforms [MLIR][TORCH] Add E2E support for aten.new_zeros, aten.new_ones op 2022-03-01 22:09:47 +05:30
Utils Revert "[LINALG] Decompose `aten.batch_norm` into `aten.native_batch_norm`" 2022-02-25 15:46:55 +05:30
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00