projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
04e8162
)
[nnvm] fix nnvm compiler build module error (#3378)
author
Howave
<myhouseng@gmail.com>
Mon, 17 Jun 2019 16:56:58 +0000
(
00:56
+0800)
committer
Tianqi Chen
<tqchen@users.noreply.github.com>
Mon, 17 Jun 2019 16:56:58 +0000
(09:56 -0700)
nnvm/python/nnvm/compiler/build_module.py
patch
|
blob
|
history
diff --git
a/nnvm/python/nnvm/compiler/build_module.py
b/nnvm/python/nnvm/compiler/build_module.py
index
30ddc7c
..
c52265f
100644
(file)
--- a/
nnvm/python/nnvm/compiler/build_module.py
+++ b/
nnvm/python/nnvm/compiler/build_module.py
@@
-148,7
+148,7
@@
def _update_shape_dtype(shape, dtype, params):
shape.update({k : v.shape for k, v in params.items()})
if isinstance(dtype, str):
for k, v in params.items():
- if v.dtype != dtype:
+ if v.dtype != dtype
and v.shape
:
raise ValueError(
"%s: dtype not expected %s vs %s" % (k, dtype, v.dtype))
else: