projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2ce1193) | patch
[TF] Fix some shape mismatches between TF and Relay (#6166)
authorlixiaoquan <radioheads@163.com>
Wed, 29 Jul 2020 17:34:32 +0000 (01:34 +0800)
committerGitHub <noreply@github.com>
Wed, 29 Jul 2020 17:34:32 +0000 (10:34 -0700)
commit760a547715e18a3495a6425a8f18c359cf107dc6
tree36f0a11b923bf822b20525fa14b784efdb39d7b6tree | snapshot
parent2ce1193aa46ed4e22f47f6c7b72b7998646a57f8commit | diff
[TF] Fix some shape mismatches between TF and Relay (#6166)

Make ndarray_size output scalar
  Make gather_nd output scalar if needed
src/relay/op/tensor/transform.cc diff | blob | history
src/relay/op/tensor/unary.cc diff | blob | history
src/relay/transforms/fold_constant.cc diff | blob | history
tests/python/frontend/tensorflow/test_forward.py diff | blob | history
tests/python/relay/test_pass_fold_constant.py diff | blob | history
topi/include/topi/transform.h diff | blob | history
topi/tests/python/test_topi_transform.py diff | blob | history
Domain: Machine Learning / ML Framework;