[Relay] Better shape inference in TensorFlow Frontend. (#3176)
authorJosh Fromm <jwfromm@uw.edu>
Fri, 17 May 2019 10:41:50 +0000 (03:41 -0700)
committerSiva <sivar.b@huawei.com>
Fri, 17 May 2019 10:41:50 +0000 (16:11 +0530)
commit246b41092981ccc4675cf8c97636745e9e59e370
tree47f940ab9450f72febf99de125759a1c24f0b322
parentac3f5bd96f05e93686f9607fff822275e685a90e
[Relay] Better shape inference in TensorFlow Frontend. (#3176)

* Some bug fixes in tensorflow graph converter and added DepthToSpace operator.

* Made DepthToSpace better comply with other function syntax.

* Added better shape inference for unusual situations.

* Lint fixes.

* Added depthtospace test.

* Added test cases for value inference and depthtospace.

* Added fill testing.

* Made comment changes and added BroadcastTo op and tests.

* Fixed underlining and unneeded opt_level forcing.

* Added _infer_value assertion that all values to infer are available in passed parameters.
python/tvm/relay/frontend/tensorflow.py
tests/python/frontend/tensorflow/test_forward.py