llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch
History
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
..
IR [MLIR][TORCH] E2E support for [ge|ceil].float, [ge|ne|gt].float_int op 2022-05-05 21:48:35 +05:30
Transforms [TORCH] Add folding of aten.detach op. 2022-05-10 21:54:45 +05:30
Utils Fix type promotion code for scalar only operations 2022-05-07 10:37:21 -04:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00