提交 35c45b33 编写于 作者: S SunAhong1993

remove print

上级 d15ee1e9
...@@ -262,7 +262,7 @@ class TFOpMapper(OpMapper): ...@@ -262,7 +262,7 @@ class TFOpMapper(OpMapper):
else: else:
inputs["shape"] = dims.name inputs["shape"] = dims.name
layer_attrs["dtype"] = string(input_value.dtype) layer_attrs["dtype"] = string(input_value.dtype)
layer_attrs["value"] = input_value.value layer_attrs["fill_value"] = input_value.value
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"paddle.full", "paddle.full",
...@@ -1246,14 +1246,12 @@ class TFOpMapper(OpMapper): ...@@ -1246,14 +1246,12 @@ class TFOpMapper(OpMapper):
y_shape = y.out_shapes[0] y_shape = y.out_shapes[0]
layer_id = self.paddle_graph.add_layer( layer_id = self.paddle_graph.add_layer(
"paddle.fluid.layers.elementwise_sub", inputs=inputs, outputs=[node.name]) "paddle.fluid.layers.elementwise_sub", inputs=inputs, outputs=[node.name])
# program.layers[layer_id].input_shapes = {"x": x_shape, "y": y_shape}
inputs = {"x": node.name, "y": node.name} inputs = {"x": node.name, "y": node.name}
x_shape = node.out_shapes[0] x_shape = node.out_shapes[0]
y_shape = node.out_shapes[0] y_shape = node.out_shapes[0]
layer_id = self.paddle_graph.add_layer( layer_id = self.paddle_graph.add_layer(
"paddle.multiply", inputs=inputs, outputs=[node.name]) "paddle.multiply", inputs=inputs, outputs=[node.name])
# program.layers[layer_id].input_shapes = {"x": x_shape, "y": y_shape}
def OneHot(self, node): def OneHot(self, node):
input = self.graph.get_node(node.layer.input[0]) input = self.graph.get_node(node.layer.input[0])
......
...@@ -178,4 +178,3 @@ class BatchNormOpt: ...@@ -178,4 +178,3 @@ class BatchNormOpt:
graph.layers[bn.id] = bn graph.layers[bn.id] = bn
graph.layers[transpose1.id] = transpose1 graph.layers[transpose1.id] = transpose1
graph.build() graph.build()
print("=============")
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册