torch-mlir/lib/Dialect/Torch
武家伟 3b3cb99ef8
Generalize canonicalization pattern for more aten.sub/div/mul/add op (#1209)
Generalize canonicalization pattern for more sub/div/mul/add op, but for AtenDivTensorModeOp in 'trunc' rounding mode, we try to fold it.
2022-08-16 13:24:08 +08:00
..
IR Generalize canonicalization pattern for more aten.sub/div/mul/add op (#1209) 2022-08-16 13:24:08 +08:00
Transforms Fix unused-variables warnings about EmbeddingBag ops (#1220) 2022-08-15 09:43:55 -07:00
Utils importer: add initial support for loading Float16 tensors (#1169) 2022-08-08 12:37:31 +08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00