torch-mlir/lib/Dialect/Torch
zjgarvey d61986cfcf
Add Decompostion for `Aten_SafeSoftmaxOp` (#3708)
Co-authored-by: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
2024-09-12 16:58:10 -05:00
..
IR [Torch] enhance fold of aten.alias (#3705) 2024-09-12 17:04:57 +08:00
Transforms Add Decompostion for `Aten_SafeSoftmaxOp` (#3708) 2024-09-12 16:58:10 -05:00
Utils [Torch] add AtenAsStridedOp in torch dialect (#3706) 2024-09-12 19:07:11 +08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00