[Relay] Add shape check for ConcatenateRel and StackRel (#3699)
authorLigeng Zhu <Lyken17@users.noreply.github.com>
Wed, 18 Sep 2019 23:03:09 +0000 (19:03 -0400)
committerZhi <5145158+zhiics@users.noreply.github.com>
Wed, 18 Sep 2019 23:03:09 +0000 (16:03 -0700)
commitcdbf4d859446f1244dbda9ff668eec806ed9213d
treef5db16af86460d9468dd76aa0438dc8ca8884b62
parentf3abb3d88f3a1145a4454649b454fe0139f19bc9
[Relay] Add shape check for ConcatenateRel and StackRel (#3699)

* [Relay] add shape check for concat

* [Relay] add shape check for stack

* add test case for shape mismatch

* [typo] add the missing assert

* fix lint errors.

* replace int with size_t.

* statically cast param->axis to size_t.

* switch to run_infer_type.

* fix checking for negative index

* add static_cast for param->axis

* merge to latest tvm

* fix lint error

* Fix an error with negative index.

* Update transform.h

* Update transform.cc
src/relay/op/tensor/transform.cc
src/relay/op/tensor/transform.h
tests/python/relay/test_op_level1.py