[Relay/TOPI][Op] Add shape op in Relay and TOPI (#2749)
authorHaichen Shen <shenhaichen@gmail.com>
Wed, 13 Mar 2019 23:14:48 +0000 (16:14 -0700)
committerLeyuan Wang <laurawly@gmail.com>
Wed, 13 Mar 2019 23:14:48 +0000 (16:14 -0700)
commitee8058069a41a80845b952a371c5beb2570f6d24
tree3067dcca9a208acb32f2d2e49aa5d62f4a8b744e
parent4d09fc4e485cda04b64a81cf149b70611a0f016b
[Relay/TOPI][Op] Add shape op in Relay and TOPI (#2749)

* Add shapeof op in topi

* Add relay shape_of op

* Add constant folding for shape_of

* Allow shape op to specify dtype

* Add mxnet converter for shape_array

* lint

* lint

* Add doc
15 files changed:
docs/api/python/topi.rst
docs/langref/relay_op.rst
include/tvm/relay/attrs/transform.h
python/tvm/relay/frontend/mxnet.py
python/tvm/relay/op/_tensor.py
python/tvm/relay/op/tensor.py
src/relay/op/tensor/unary.cc
src/relay/pass/fold_constant.cc
tests/python/frontend/mxnet/test_forward.py
tests/python/relay/test_op_level10.py
tests/python/relay/test_pass_fold_constant.py
topi/include/topi/transform.h
topi/python/topi/transform.py
topi/src/topi.cc
topi/tests/python/test_topi_transform.py