提交 a9bb2791 编写于 作者: M mamingjie-China

update

上级 c4714f3f
...@@ -1026,3 +1026,13 @@ class TFOpMapperNHWC(OpMapper): ...@@ -1026,3 +1026,13 @@ class TFOpMapperNHWC(OpMapper):
inputs=inputs, inputs=inputs,
outputs=[node.name], outputs=[node.name],
**attr) **attr)
def SquaredDifference(self, node):
x = self.graph.get_node(node.layer.input[0], copy=True)
y = self.graph.get_node(node.layer.input[1], copy=True)
inputs = {"x": x.name, "y": y.name}
program.add_layer(
"fluid.layers.elementwise_sub", inputs=inputs, outputs=[node.name])
inputs = {"x": node.name, "y": node.name}
program.add_layer(
"fluid.layers.elementwise_mul", inputs=inputs, outputs=[node.name])
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册