提交 6fa0a8ae authored 作者: James Bergstra's avatar James Bergstra

test shapefeature - removed infer_shape that depended on node.env

上级 9184ad40
......@@ -1033,8 +1033,8 @@ class test_shapeoptimizer(unittest.TestCase):
return Apply(self, [x], [x.type()])
def perform(self, node, (x,), (out,)):
out[0] = x.copy()
def infer_shape(self, node, (xshp,)):
return node.env.shape_feature.default_infer_shape(node, (xshp,))
#def infer_shape(self, node, (xshp,)):
#return [tuple([self.shape_i(i)(r) for i in xrange(r.ndim)])]
identity_noshape = IdentityNoShape()
class IdentityShape(Op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论