torch-mlir/lib/Dialect/Torch
Yuanqiang Liu 8814d0ae64
[Torch] emit aten.dot and canonicalize it to aten.matmul (#3361)
* canonicalize `aten.dot` to `aten.matmul`
2024-05-18 22:45:14 +08:00
..
IR [Torch] emit aten.dot and canonicalize it to aten.matmul (#3361) 2024-05-18 22:45:14 +08:00
Transforms [Torch] emit aten.dot and canonicalize it to aten.matmul (#3361) 2024-05-18 22:45:14 +08:00
Utils [Torch] Decompose AtenMaskedScatterOp (#3353) 2024-05-16 15:27:25 +08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00