提交 ac1660e5 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix crash recently introduced

上级 aa2a430b
...@@ -4469,7 +4469,8 @@ class Reshape(Op): ...@@ -4469,7 +4469,8 @@ class Reshape(Op):
# to give a positive value. # to give a positive value.
# To avoid optimization complexity, we avoid checking # To avoid optimization complexity, we avoid checking
# for the case when there are two or more '-1' values. # for the case when there are two or more '-1' values.
rest_size = (mul(*ishapes[0]) // -mul(*new_dims)) if self.ndim:
rest_size = (mul(*ishapes[0]) // -mul(*new_dims))
return [tuple([switch(eq(new_dims[i], -1), return [tuple([switch(eq(new_dims[i], -1),
rest_size, rest_size,
new_dims[i]) new_dims[i])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论