未验证 提交 5480c7dc 编写于 作者: J Jason 提交者: GitHub

Update tf_op_mapper.py

上级 8f1f12ba
...@@ -818,7 +818,7 @@ class TFOpMapper(OpMapper): ...@@ -818,7 +818,7 @@ class TFOpMapper(OpMapper):
node.fluid_code.add_layer("range", node.fluid_code.add_layer("range",
inputs=inputs, inputs=inputs,
output=node, output=node,
param_attr=None) param_attr=attr)
def Mean(self, node): def Mean(self, node):
input = self.graph.get_node(node.layer.input[0], copy=True) input = self.graph.get_node(node.layer.input[0], copy=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册