torch-mlir/python/torch_mlir
Prashant Kumar 12b3af70d3 [TORCH] Add folding of aten.detach op.
`aten.detach` op is folded and returns the first operand since it's an
identity function(kind of identity just remove the has_grad attribute).
2022-05-10 21:54:45 +05:30
..
dialects [TORCH] Add folding of aten.detach op. 2022-05-10 21:54:45 +05:30
eager_mode Update torch_mlir_tensor.py 2022-05-07 21:46:10 -04:00
__init__.py Add tracing suport to `torch_mlir.compile`. 2022-05-03 09:08:40 -07:00
compiler_utils.py Add a new `torch_mlir.compile` method. 2022-04-20 10:06:01 -07:00