torch-mlir/python
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
..
test Pure-Python FX importer. (#2098) 2023-05-12 00:46:33 -05:00
torch_mlir [Torch Dialect] Add split.tensor support + recompose rules (#2102) 2023-05-23 12:43:33 -07:00
torch_mlir_e2e_test [Torch Dialect] Add split.tensor support + recompose rules (#2102) 2023-05-23 12:43:33 -07:00
CMakeLists.txt Pure-Python FX importer. (#2098) 2023-05-12 00:46:33 -05:00
TorchMLIRModule.cpp Miscellaneous fixes for Windows builds (#1376) 2022-09-29 12:07:43 -05:00