torch-mlir/lib/Dialect/Torch
Zhekun Zhang a426363b7d
[Torch Dialect] Add split.tensor support + recompose rules (#2102)
* add split.tensor support + recompose rules

* add e2e test

* address comments

* address comments

* erase op in recomposeOp

---------

Co-authored-by: zhekun.zhang <zhekun.zhang@bytedance.com>
2023-05-23 12:43:33 -07:00
..
IR [Torch Dialect] Enforce signless attribute for ConstantIntOp (#2078) 2023-05-22 19:21:34 -05:00
Transforms [Torch Dialect] Add split.tensor support + recompose rules (#2102) 2023-05-23 12:43:33 -07:00
Utils Add complex types support with basic complex ops. 2023-05-11 21:29:07 +05:30
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00