1Vh,dZddlmZdZdZdZdZy)zUtilities for graph plugin.) graph_pb2c|d|S)N/)prefixop_names T/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/plugins/graph/graph_util.py_prefixed_op_namer s g &&c|d|S)aqReturns function name prefixed with `prefix`. For function libraries, which are often created out of autographed Python function, are factored out in the graph vis. They are grouped under a function name which often has a shape of `__inference_[py_func_name]_[numeric_suffix]`. While it does not have some unique information about which graph it is from, creating another wrapping structure with graph prefix and "/" is less than ideal so we join the prefix and func_name using underscore. TODO(stephanwlee): add business logic to strip "__inference_" for more user friendlier name _r)r func_names r _prefixed_func_namersi ((r c|jD]}|jj}|j|t||j|_|j Dcgc]}t||c}|j dd|j dk(s|jdst||jdjj|jdj_|jjD]f}|jjj}|j|t||jj|j_h|jjD]m}|jjj} | j|t|| j| _t|| j| _oycc}w)NPartitionedCallf)nodeaddCopyFromr nameinputopattrrfunclibraryfunction signaturegradient function_name gradient_func) r graph_to_adddestination_graphrnew_node input_namernew_funcr new_gradients r _add_with_prepended_namesr'*s!!$))--/$)&$))< DHJJ 6@ fj 1 q ;;+ + c0B+> c"'',,,HMM#  # # ("$$-- $,,5599;$"5 H&&++#  !((11  (0099==? h'%8   & && "&9   & && "  ) s$G8ct|dk(r|dSt|dk(rtjStj}|djjr(|jj |djt |D]P\}}|jj|jjk7r tdtd|dzz||R|S)aMerges GraphDefs by adding unique prefix, `graph_{ind}`, to names. All GraphDefs are expected to be of TensorBoard's. When collecting graphs using the `tf.summary.trace` API, node names are not guranteed to be unique. When non-unique names are not considered, it can lead to graph visualization showing them as one which creates inaccurate depiction of the flow of the graph (e.g., if there are A -> B -> C and D -> B -> E, you may see {A, D} -> B -> E). To prevent such graph, we checked for uniquenss while merging but it resulted in https://github.com/tensorflow/tensorboard/issues/1929. To remedy these issues, we simply "apply name scope" on each graph by prefixing it with unique name (with a chance of collision) to create unconnected group of graphs. In case there is only one graph def passed, it returns the original graph_def. In case no graph defs are passed, it returns an empty GraphDef. Args: graph_defs: TensorBoard GraphDefs to merge. Returns: TensorBoard GraphDef that merges all graph_defs with unique prefixes. Raises: ValueError in case GraphDef versions mismatch. rz/Cannot combine GraphDefs of different versions.zgraph_%d) lenrGraphDefversionsproducerr enumerate ValueErrorr') graph_defs dst_graph_defindex graph_defs r merge_graph_defsr4Ps: :!!} ZA !!##&&(M!}&&'' 1 (>(>?%j1 y  ! ! * *i.@.@.I.I INO O! %!) $     r N)__doc__tensorboard.compat.protorr rr'r4rr r r7s"".')$# L1r