_transform.py
20.8 KB
-
[Relay][Pass]Improve memory_allocation pass to support multiple i/o dynamic kernels (#4595) · 9c638f06
* Add more shape funcs * Fix test * Enhance test_any_concat * Fix pylint * Minor fix test * Fix pylint * Minor refactor * Add test any for elemwise
Yao Wang committed