提交 7c5fb3dd 编写于 作者: C Channingss

onnx_shape_inference support python2, delete repeate OP:greater

上级 f3e37fcf
...@@ -267,9 +267,9 @@ class SymbolicShapeInference: ...@@ -267,9 +267,9 @@ class SymbolicShapeInference:
if pending_nodes and self.verbose_ > 0: if pending_nodes and self.verbose_ > 0:
print('SymbolicShapeInference: orphaned nodes discarded: ') print('SymbolicShapeInference: orphaned nodes discarded: ')
print( for n in pending_nodes:
* [n.op_type + ': ' + n.output[0] for n in pending_nodes], print(n.op_type + ': ' + n.output[0])
sep='\n')
if input_shapes is not None: if input_shapes is not None:
for input_name, shape in input_shapes.items(): for input_name, shape in input_shapes.items():
for idx in range(len(self.out_mp_.graph.input)): for idx in range(len(self.out_mp_.graph.input)):
......
...@@ -492,16 +492,6 @@ class OpSet9(): ...@@ -492,16 +492,6 @@ class OpSet9():
node.fluid_code.add_layer( node.fluid_code.add_layer(
'hard_shrink', inputs=val_x, output=node, param_attr=attr) 'hard_shrink', inputs=val_x, output=node, param_attr=attr)
def Greater(self, node):
val_x = self.graph.get_input_node(node, idx=0, copy=True)
val_y = self.graph.get_input_node(node, idx=1, copy=True)
node.fluid_code.add_layer(
'greater_than',
inputs={'x': val_x,
'y': val_y},
output=node,
param_attr=None)
@print_mapping_info @print_mapping_info
def Constant(self, node): def Constant(self, node):
val_output = self.graph.get_node(node.layer.output[0], copy=True) val_output = self.graph.get_node(node.layer.output[0], copy=True)
...@@ -1160,7 +1150,6 @@ class OpSet9(): ...@@ -1160,7 +1150,6 @@ class OpSet9():
'y': cast_condition}, 'y': cast_condition},
output=mul_val_x, output=mul_val_x,
param_attr=None) param_attr=None)
mul_val_y = val_y.layer_name + '_mul' mul_val_y = val_y.layer_name + '_mul'
node.fluid_code.add_layer( node.fluid_code.add_layer(
"elementwise_mul", "elementwise_mul",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册