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

Fix crash recently introduced

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