[TOPI] implement pool3d op (#4478)
authoroptima2005 <56945758+optima2005@users.noreply.github.com>
Thu, 12 Dec 2019 06:06:20 +0000 (14:06 +0800)
committermasahi <masahi129@gmail.com>
Thu, 12 Dec 2019 06:06:20 +0000 (15:06 +0900)
commit41959ed26b2629b48148a84d0959a0c2b75dee28
tree841babe3b00b42d949f3d5fe328d03f07ba044d7
parent8c2d4f6504b61a727b1391ea2424022d7f12bf2e
[TOPI] implement pool3d op (#4478)

* [TOPI] implement pool3d op

* use PoolInferCorrectLayout for both 2d and 3d pooling

* unify MakeMaxPool and MakeAvgPool
include/tvm/relay/attrs/nn.h
src/relay/op/nn/pooling.cc
topi/include/topi/nn/pooling.h
topi/python/topi/nn/pooling.py
topi/src/topi.cc
topi/tests/python/test_topi_pooling.py