Register Shape Func for Some Operators to Handle Dynamic Shapes (#5955)
authorSiyuan Li <siyuanli.s.c@gmail.com>
Thu, 23 Jul 2020 20:23:49 +0000 (04:23 +0800)
committerGitHub <noreply@github.com>
Thu, 23 Jul 2020 20:23:49 +0000 (13:23 -0700)
commitfe761964029b1094b00925fe577bc8bf5f9f2d66
tree4f7d3e109dca136b5c4e150a800cbcecd7f5a0a9
parent9d34eaaa34f14fbfa6ad3540cd93c2ca08571207
Register Shape Func for Some Operators to Handle Dynamic Shapes (#5955)

* Register Shape Func for Floor Operator

Register the shape function for `floor` operator. Otherwise, a bug will happen when input of floor is any.

* Register shape func for log

* add shape function for crop_and_size

* change import location

* add mirror_pad shape function

* add test cases for crop_and_resize and mirror_pad shape funcs

* support different layout

* fix pylint error

* fix pylint error

* add test for nchw layout

* block nchw test

* test for nchw

* use tvm.testing.assert_allclose instead

Co-authored-by: lisiyuan <lisiyuan@nucflow>
python/tvm/relay/op/_tensor.py
python/tvm/relay/op/image/_image.py
python/tvm/relay/op/nn/_nn.py
tests/python/relay/test_any.py