llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python/torch_mlir
History
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
..
dialects [MLIR][TORCH] Add E2E support for torch.arange op 2021-12-27 22:45:48 +05:30