[Relay][Op] Adaptive pooling (#3085)
authorYao Wang <kevinthesunwy@gmail.com>
Thu, 9 May 2019 00:21:41 +0000 (17:21 -0700)
committerHaichen Shen <shenhaichen@gmail.com>
Thu, 9 May 2019 00:21:41 +0000 (17:21 -0700)
commit147ea3b0ca147b527086228d524a2f68f872112d
tree9d87bff5ecb1a2f3d6179e5da7b0ecd229d6a044
parenta6d04b8daaa1e75b00b61755260f8ea17f07ba7c
[Relay][Op] Adaptive pooling (#3085)

* Add topi adaptive_pool

* Use adaptive_pool to compute global_pool

* Add relay adaptive pool2d

* Fix lint

* Fix typo

* Minor change

* Change support level to 10

* Add contrib

* Remove global pool schedule

* Add contrib module

* Fix lint

* Update doc

* Update doc
29 files changed:
docs/api/python/topi.rst
docs/langref/relay_op.rst
include/tvm/relay/attrs/nn.h
nnvm/python/nnvm/top/nn.py
python/tvm/relay/__init__.py
python/tvm/relay/contrib.py [new file with mode: 0644]
python/tvm/relay/frontend/mxnet.py
python/tvm/relay/op/__init__.py
python/tvm/relay/op/contrib/__init__.py [new file with mode: 0644]
python/tvm/relay/op/contrib/_contrib.py [new file with mode: 0644]
python/tvm/relay/op/contrib/_make.py [new file with mode: 0644]
python/tvm/relay/op/contrib/contrib.py [new file with mode: 0644]
python/tvm/relay/op/nn/_nn.py
src/relay/op/nn/pooling.cc
tests/python/frontend/mxnet/test_forward.py
tests/python/relay/test_op_level10.py
tests/python/relay/test_op_level2.py
topi/include/topi/nn/pooling.h
topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/pooling.py
topi/python/topi/generic/nn.py
topi/python/topi/hls/nn.py
topi/python/topi/nn/pooling.py
topi/python/topi/opengl/__init__.py
topi/python/topi/opengl/pooling.py
topi/python/topi/x86/__init__.py
topi/python/topi/x86/pooling.py
topi/src/topi.cc
topi/tests/python/test_topi_pooling.py