torch-mlir/lib/Dialect/Torch
Xinyu Yang 42a16fa912
[Torch] Support Aten_CastFloatOp. (#3115)
By canonicalize Aten_CastFloatOp into AtenToDtypeOp
2024-04-09 11:06:53 +08:00
..
IR [Torch] Support Aten_CastFloatOp. (#3115) 2024-04-09 11:06:53 +08:00
Transforms [Torch] Support Aten_CastFloatOp. (#3115) 2024-04-09 11:06:53 +08:00
Utils Adds accumulator types in TorchToLinalg for `AtenMmOp` and `AtenConvolutionOp` (#3027) 2024-03-14 16:40:40 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00