[TOPI][Relay] max_pool2d & avg_pool2d gradient (#3601)
authorWuwei Lin <wuwei@apache.org>
Wed, 24 Jul 2019 18:30:46 +0000 (02:30 +0800)
committerTianqi Chen <tqchen@users.noreply.github.com>
Wed, 24 Jul 2019 18:30:46 +0000 (11:30 -0700)
commit5c41003726216207013d76f567e101f5f75c74ae
treeefd202caf530372302f597fdeecc1114852a0a93
parent440df0aa819d8bbdc2686c7ef4a4d263e0f2b2c8
[TOPI][Relay] max_pool2d & avg_pool2d gradient (#3601)
15 files changed:
python/tvm/relay/op/_tensor_grad.py
python/tvm/relay/op/nn/_nn.py
python/tvm/relay/op/nn/nn.py
python/tvm/relay/op/op_attrs.py
src/relay/op/nn/pooling.cc
topi/include/topi/detail/ravel_unravel.h
topi/include/topi/nn/pooling.h
topi/include/topi/reduction.h
topi/include/topi/transform.h
topi/python/topi/generic/nn.py
topi/python/topi/nn/pooling.py
topi/python/topi/testing/__init__.py
topi/python/topi/testing/pool_grad_python.py [new file with mode: 0644]
topi/src/topi.cc
topi/tests/python/test_topi_pooling.py