[Relay][Op]Support symbolic TopK, Ones, Zeros and Full (#5459)
authorYao Wang <kevinthesunwy@gmail.com>
Tue, 26 May 2020 01:09:44 +0000 (18:09 -0700)
committerGitHub <noreply@github.com>
Tue, 26 May 2020 01:09:44 +0000 (18:09 -0700)
commitcc79591f5e0f05955fd3180b1975e4344b532345
treec47dcbb47b07d3414aa8452237b7a2514188a63b
parent7b74a8672e1e40e7541c0007d8628586c62277e8
[Relay][Op]Support symbolic TopK, Ones, Zeros and Full (#5459)

* Support symbolic TopK, Ones, Zeros and Full

* Fix pylint

* Add docstring for topk shape func

* Fix grad

* Fix lazy_gradient_init

* Fix parser

* Fix print ir text

* Fix lint

* Improve pattern_util

* Fix topk

* Fix build

* Use Optional for attribute

* Fix clang-format

* Minot fix

* Fix pylint

* Fix build warning

* Fix parser

* Move ToScalar

* Fix lint

* Fix lint

* Make topk shape func as data independent when k is constant.

* Fix lint

* Minor fix
22 files changed:
include/tvm/relay/attrs/algorithm.h
include/tvm/relay/attrs/transform.h
include/tvm/runtime/ndarray.h
python/tvm/relay/_parser.py
python/tvm/relay/op/_algorithm.py
python/tvm/relay/op/_tensor.py
python/tvm/relay/op/_tensor_grad.py
python/tvm/relay/op/_transform.py
python/tvm/relay/op/algorithm.py
python/tvm/relay/op/strategy/generic.py
python/tvm/relay/op/tensor.py
python/tvm/relay/op/transform.py
src/relay/analysis/util.cc
src/relay/op/algorithm/topk.cc
src/relay/op/image/resize.cc
src/relay/op/tensor/transform.cc
src/relay/op/tensor/transform.h
src/relay/qnn/util.cc
src/relay/transforms/lazy_gradient_init.cc
src/relay/transforms/pattern_util.h
tests/python/relay/test_any.py
topi/python/topi/sort.py