提交 7626cd8a 编写于 作者: S SunAhong1993

rename

上级 d4223528
...@@ -76,7 +76,7 @@ class TFOpMapper(OpMapper): ...@@ -76,7 +76,7 @@ class TFOpMapper(OpMapper):
'LessEqual': 'paddle.less_equal', 'LessEqual': 'paddle.less_equal',
'GreaterEqual': 'paddle.greater_equal', 'GreaterEqual': 'paddle.greater_equal',
'Mul': 'paddle.multiply', 'Mul': 'paddle.multiply',
'FloorDiv': 'fluid.layers.elementwise_floordiv' 'FloorDiv': 'paddle.floor_divide'
} }
def __init__(self, decoder): def __init__(self, decoder):
...@@ -1039,7 +1039,7 @@ class TFOpMapper(OpMapper): ...@@ -1039,7 +1039,7 @@ class TFOpMapper(OpMapper):
if data_format == "NHWC": if data_format == "NHWC":
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
kernel="fluid.layers.transpose", kernel="paddle.transpose",
inputs={"x": node.name}, inputs={"x": node.name},
outputs=[node.name], outputs=[node.name],
perm=[0, 2, 3, 1]) perm=[0, 2, 3, 1])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册