torch-mlir/python/torch_mlir/dialects
Vivek Khandelwal 4486de5ef3 [MLIR][TORCH] Add E2E support for torch.arange op
This commit adds lowering of `aten.arange.start_step` op.
This commit decomposes `aten.arange` and `aten.arange.start` into
`aten.arange.start_step` op.

Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2021-12-27 22:45:48 +05:30
..
torch [MLIR][TORCH] Add E2E support for torch.arange op 2021-12-27 22:45:48 +05:30
TorchBinding.td Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00