test_topi_pooling.py
15.4 KB
-
[TOPI] implement pool3d op (#4478) · 41959ed2* [TOPI] implement pool3d op * use PoolInferCorrectLayout for both 2d and 3d pooling * unify MakeMaxPool and MakeAvgPool
optima2005 committed
* [TOPI] implement pool3d op * use PoolInferCorrectLayout for both 2d and 3d pooling * unify MakeMaxPool and MakeAvgPool