提交 0b0335f9 编写于 作者: S SunAhong1993

add beg

上级 7794f4d3
...@@ -248,6 +248,15 @@ class TFOpMapper(OpMapper): ...@@ -248,6 +248,15 @@ class TFOpMapper(OpMapper):
inputs={"x": input.name}, inputs={"x": input.name},
outputs=[node.name], outputs=[node.name],
perm=perm) perm=perm)
def Neg(self, node):
input = self.graph.get_input_node(node, 0)
self.paddle_graph.add_layer(
"paddle.scale",
inputs={"x": input.name},
outputs=[node.name],
scale=-1)
def Fill(self, node): def Fill(self, node):
dims = self.graph.get_input_node(node, 0) dims = self.graph.get_input_node(node, 0)
......
...@@ -304,6 +304,15 @@ class TFOpMapper(OpMapper): ...@@ -304,6 +304,15 @@ class TFOpMapper(OpMapper):
inputs={"x": node.name}, inputs={"x": node.name},
outputs=[node.name], outputs=[node.name],
perm=[0, 2, 3, 1]) perm=[0, 2, 3, 1])
def Neg(self, node):
input = self.graph.get_input_node(node, 0)
self.paddle_graph.add_layer(
"paddle.scale",
inputs={"x": input.name},
outputs=[node.name],
scale=-1)
def MaxPool(self, node): def MaxPool(self, node):
input = self.graph.get_node(node.layer.input[0]) input = self.graph.get_node(node.layer.input[0])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册