[topi][relay] Add operation gather to relay. (#5716)
authornotoraptor <notoraptor@users.noreply.github.com>
Fri, 12 Jun 2020 15:24:56 +0000 (11:24 -0400)
committerGitHub <noreply@github.com>
Fri, 12 Jun 2020 15:24:56 +0000 (08:24 -0700)
commitf677af0e9b540240e51b63df463edfcc761d1442
tree5e54050f5a033095f2072fe8e39460ba45d738a4
parentc9f2e2f73d6160abbd314d4c647437b5f49336df
[topi][relay] Add operation gather to relay. (#5716)
14 files changed:
docs/api/python/topi.rst
docs/langref/relay_op.rst
include/tvm/relay/attrs/transform.h
python/tvm/relay/op/_transform.py
python/tvm/relay/op/op_attrs.py
python/tvm/relay/op/transform.py
src/relay/op/tensor/transform.cc
tests/python/relay/test_op_level3.py
topi/include/topi/transform.h
topi/python/topi/testing/__init__.py
topi/python/topi/testing/gather_python.py [new file with mode: 0644]
topi/python/topi/transform.py
topi/src/transform.cc
topi/tests/python/test_topi_transform.py