[Relay][Topi][Op]Advanced indexing (#6388)
authorYao Wang <kevinthesunwy@gmail.com>
Fri, 11 Sep 2020 06:09:45 +0000 (23:09 -0700)
committerGitHub <noreply@github.com>
Fri, 11 Sep 2020 06:09:45 +0000 (23:09 -0700)
commit122811137a6fe3e787d5174a36d99eebefb479d0
tree5f81497ed298c65a011f95f27b362a2d27204bb9
parent355720ef48c64e8567303b3d895ceb84c6d63d26
[Relay][Topi][Op]Advanced indexing (#6388)

* Add Relay adv_index op

* Support single index tensor dynamic shape

* Support more dynamic index

* Fix lint

* Minor fix for comment

* Fix lint

* Fix lint

* Fix test

* Fix
include/tvm/topi/transform.h
python/tvm/relay/frontend/pytorch.py
python/tvm/relay/op/_transform.py
python/tvm/relay/op/transform.py
python/tvm/topi/transform.py
src/relay/op/tensor/transform.cc
src/topi/transform.cc
tests/python/relay/test_any.py
tests/python/relay/test_op_level3.py
tests/python/topi/python/test_topi_transform.py