test_topi_pooling.py
17.2 KB
-
[Relay, TOPI] Refactor Adaptive pool and add 3d support (#5049) · 7c5ff508
* add stub for nd impl * refactored indices compute * refactored divide step * remove unused variables, add doc * fix lint * add relay op def * add python registration * refactor topi test * update relay tests, but test result is weird * workaround for weird bug * add relay adaptive pool 3d test * add topi tests * update doc for 3d * typo fix * fix lint * add more tests including NDHWC
masahi committed