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

small bug fix.

上级 878c8f1c
......@@ -426,7 +426,7 @@ class ShapeFeature(object):
_logger.error('Failed to infer_shape from Op %s (i_shapes=%s): %s %s'% (node.op,
[self.shape_of[r] for r in node.inputs],
type(e), str(e)))
o_shapes = default_infer_shape(node, [self.shape_of[r] for r in node.inputs])
o_shapes = self.default_infer_shape(node, [self.shape_of[r] for r in node.inputs])
# this is packed information
# an element of o_shapes is either None or a tuple
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论