torch-mlir/lib/Dialect/Torch
Vidush Singhal b70548edff
Add decomposition and E2E support for Aten_EmbeddingBag (#1137)
* Add decomposition and E2E support for Aten_EmbeddingBag
2022-08-08 18:56:49 -04:00
..
IR [MLIR][TORCH] Add support for multiple indexing tensors for aten.index.Tensor (#1097) 2022-07-28 19:00:02 -04:00
Transforms Add decomposition and E2E support for Aten_EmbeddingBag (#1137) 2022-08-08 18:56:49 -04: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