-
[Lang] Layout in TVM node system (#2509) · ee79703c
* move layout.h & layout.cc from relay to tvm * change ConvertLayout in relay to bijectiveLayout->Forward/backward * add first test case * add LayoutAxis * add LayoutAxis struct and compiles * simplify BijectiveLayout rule consturct * polish func name for Layout, move impl to .cc, remove Layout::defined(), add defined() checker * partially add layout py support * add layout test cases * add doc for tvm.layout & tvm.bijective_layout * fix lint * fix lint * fix layout name generation bug * fix layout typo * address comments and add topi.layout_transform * layout.h->data_layout.h, test_lang_layout.py->test_lang_data_layout.py
Yizhi Liu committed
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
binary.cc | Loading commit data... | |
reduce.cc | Loading commit data... | |
transform.cc | Loading commit data... | |
unary.cc | Loading commit data... |