提交 d4531ad4 编写于 作者: J jiangjiajun

little modify

上级 24e799a9
......@@ -297,10 +297,6 @@ class TFOpMapper(OpMapper):
perm = perm.value.tolist()
if perm == [0, 3, 1, 2] and input.data_format == "NHWC":
# node.fluid_code.add_layer("assign",
# inputs=input,
# output=node,
# param_attr=None)
input_name = input.layer_name
if hasattr(input, "index"):
input_name = input_name + "[{}]".format(input.index)
......@@ -1006,20 +1002,6 @@ class TFOpMapper(OpMapper):
output=node,
param_attr=attr)
# def FloorDiv(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, 'y': y}
# node.fluid_code.add_layer("elementwise_div",
# inputs=inputs,
# output=node,
# param_attr=None)
# node.fluid_code.add_layer("floor",
# inputs=node,
# output=node,
# param_attr=None)
def Split(self, node):
dim = self.graph.get_node(node.layer.input[0], copy=True)
input = self.graph.get_node(node.layer.input[1], copy=True)
......
......@@ -852,8 +852,6 @@ class TFOpMapperNHWC(OpMapper):
input = self.graph.get_node(node.layer.input[0], copy=True)
begin = self.graph.get_node(node.layer.input[1], copy=True)
size = self.graph.get_node(node.layer.input[2], copy=True)
# assert begin.layer_type == "Const"
# assert size.layer_type == "Const"
self.add_omit_nodes(begin.layer_name, node.layer_name)
self.add_omit_nodes(size.layer_name, node.layer_name)
if begin.layer_type == "Const":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册