Add More Shape Functions (#4179)
authorYao Wang <kevinthesunwy@gmail.com>
Mon, 11 Nov 2019 23:46:29 +0000 (15:46 -0800)
committerHaichen Shen <shenhaichen@gmail.com>
Mon, 11 Nov 2019 23:46:29 +0000 (15:46 -0800)
commit62521453a03d73c90900ba08ced6af06ee7f543a
tree761d409c18201418b9e725aee03eafec1a2ef684
parent10b77ef37adde1ad489e992fad87dc6069deacb1
Add More Shape Functions (#4179)

* Add shape functions

* Fix get_const_tuple

* Fix cpplint

* Fix pylint

* Fix pylint

* rebase and fix

* Check Any for infer type

* Fix expand_dim shape func for zero rank input

* Fix pooling infer type

* Address comment

* Register layout transform attr
19 files changed:
python/tvm/autotvm/task/task.py
python/tvm/autotvm/util.py
python/tvm/relay/op/_reduce.py
python/tvm/relay/op/_tensor.py
python/tvm/relay/op/_transform.py
python/tvm/relay/op/nn/_nn.py
src/lang/data_layout.cc
src/relay/op/nn/convolution.cc
src/relay/op/nn/convolution.h
src/relay/op/nn/nn.cc
src/relay/op/nn/pad.cc
src/relay/op/nn/pooling.cc
src/relay/op/tensor/reduce.cc
src/relay/op/tensor/transform.cc
tests/python/relay/test_any.py
topi/include/topi/nn/flatten.h
topi/python/topi/util.py
topi/python/topi/x86/conv2d.py
topi/python/topi/x86/dense.py