[Relay, TOPI] Deformable conv2d (#2908)
authorWuwei Lin <vincentl13x@gmail.com>
Fri, 29 Mar 2019 12:38:56 +0000 (20:38 +0800)
committermasahi <masahi129@gmail.com>
Fri, 29 Mar 2019 12:38:56 +0000 (21:38 +0900)
commit608cdeeb4852dfb340a5a9967cfa413854c9692c
tree211b6679bb53cbcb6aca2700aeed3ff2d5f4ac34
parent82e868a4be69342242c10de9567c81394a785bd2
[Relay, TOPI]  Deformable conv2d (#2908)

* [Relay, TOPI] Add deformable conv2d

* Moved to op level2

* Fix lint

* Moved to level2 & bug fix

* Update comments

* Disabled flaky test of conv2d
17 files changed:
include/tvm/relay/attrs/nn.h
python/tvm/autotvm/task/relay_integration.py
python/tvm/autotvm/task/topi_integration.py
python/tvm/relay/frontend/mxnet.py
python/tvm/relay/op/nn/_nn.py
python/tvm/relay/op/nn/nn.py
src/relay/op/nn/convolution.cc
tests/python/relay/test_op_level5.py
topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/deformable_conv2d.py [new file with mode: 0644]
topi/python/topi/generic/nn.py
topi/python/topi/nn/__init__.py
topi/python/topi/nn/deformable_conv2d.py [new file with mode: 0644]
topi/python/topi/testing/__init__.py
topi/python/topi/testing/deformable_conv2d_nchw_python.py [new file with mode: 0644]
topi/tests/python/test_topi_conv2d_nchw.py
topi/tests/python/test_topi_deformable_conv2d.py [new file with mode: 0644]