提交 8267bcc5 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Slightly more detailed error

上级 788b52f6
...@@ -382,7 +382,7 @@ class ShapeFeature(object): ...@@ -382,7 +382,7 @@ class ShapeFeature(object):
elif s_i.type == T.lscalar: elif s_i.type == T.lscalar:
return s_i return s_i
else: else:
raise TypeError('Unsupported shape element', s_i) raise TypeError('Unsupported shape element', s_i, s_i.type)
def set_shape(self, r, s): def set_shape(self, r, s):
assert r not in self.shape_of assert r not in self.shape_of
...@@ -504,7 +504,6 @@ def local_shape_i_lift(node): ...@@ -504,7 +504,6 @@ def local_shape_i_lift(node):
if inode and isinstance(inode.op, Elemwise) and inode.op.scalar_op.nin==1: if inode and isinstance(inode.op, Elemwise) and inode.op.scalar_op.nin==1:
return node.env.shape_feature.shape_of[input] return node.env.shape_feature.shape_of[input]
register_canonicalize(local_shape_i_lift) register_canonicalize(local_shape_i_lift)
register_specialize(local_shape_i_lift) register_specialize(local_shape_i_lift)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论