[Relay, Topi][OP] Correlation (#5628)
authorWuwei Lin <wuwei@apache.org>
Fri, 22 May 2020 06:00:33 +0000 (02:00 -0400)
committerGitHub <noreply@github.com>
Fri, 22 May 2020 06:00:33 +0000 (15:00 +0900)
commite55f9ff115e2e9364e8ed8b3eeb44dbbc1894eb1
tree3540b864133f539d1fe5dfa9098335f1bd9ffa13
parent4bde554109c1dce6044eaa10b62d4aad96352a9b
[Relay, Topi][OP] Correlation (#5628)

* [Relay,Topi] Correlation

* fix

* move

* typo

* Update test_topi_correlation.py
18 files changed:
include/tvm/relay/attrs/nn.h
python/tvm/relay/frontend/mxnet.py
python/tvm/relay/op/nn/_nn.py
python/tvm/relay/op/nn/nn.py
python/tvm/relay/op/op_attrs.py
python/tvm/relay/op/strategy/cuda.py
python/tvm/relay/op/strategy/generic.py
src/relay/op/nn/correlation.cc [new file with mode: 0644]
tests/python/frontend/mxnet/test_forward.py
tests/python/relay/test_op_level2.py
topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/correlation.py [new file with mode: 0644]
topi/python/topi/generic/nn.py
topi/python/topi/nn/__init__.py
topi/python/topi/nn/correlation.py [new file with mode: 0644]
topi/python/topi/testing/__init__.py
topi/python/topi/testing/correlation_nchw_python.py [new file with mode: 0644]
topi/tests/python/test_topi_correlation.py [new file with mode: 0644]