Move tensor name restoring transformation after output normalization (#9904)

This commit is contained in:
Anton Chetverikov 2022-01-27 17:41:58 +03:00 committed by GitHub
parent 578afb9bb8
commit 05684b7cd0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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()