Move tensor name restoring transformation after output normalization (#9904)
This commit is contained in:
parent
578afb9bb8
commit
05684b7cd0
@ -410,12 +410,12 @@ def copy_graph_with_ops(graph: Graph) -> Graph:
|
||||
if op.has_and_set('rt_info'):
|
||||
op.out_node(0)['rt_info'] = op.rt_info
|
||||
|
||||
restore_tensor_names(op)
|
||||
|
||||
# operations postprocessing with some special types
|
||||
if op.soft_get('type') in postprocessing_op_nodes:
|
||||
postprocessing_op_nodes[op.type](op)
|
||||
|
||||
restore_tensor_names(op)
|
||||
|
||||
# clean up graph to shape inference
|
||||
new_graph.clean_up()
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user