&Vf.dZddlmZdZdZdZdZdS)zUtilities for graph plugin.) graph_pb2c|d|S)N/)prefixop_names a/var/www/html/software/conda/lib/python3.11/site-packages/tensorboard/plugins/graph/graph_util.py_prefixed_op_namer sffgg &&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_namersffii ((r cB|jD]}|j}||t|j|_fd|jD|jdd<|jdkrG|jdr:t|jdj j|jdj _|j j D]Y}|j j }||t|j j|j _Z|j j D]i}|j j }||t|j|_t|j|_jdS)Nc0g|]}t|Sr)r ).0 input_namers r z-_add_with_prepended_names../s1   6@ fj 1 1   r PartitionedCallf)nodeaddCopyFromr nameinputopattrrfunclibraryfunction signaturegradient function_name gradient_func) r graph_to_adddestination_graphrnew_nodernew_funcr" new_gradients ` r _add_with_prepended_namesr**s!$)--//$)&$)<<     DHJ   qqq ;+ + + c0B ++> c"',,,HM#  # ( $-  $,599;;$"5 H&+# # !(1   (09==?? h'''%8   && & "&9   && & ""   r ct|dkr|dSt|dkrtjStj}|djjr%|j|djt |D]E\}}|jj|jjkrtdtd|dzz||F|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_defsr7Ps: :!!} ZA  !###&((M!}&@'' 1 (>???%j11  y  ! *i.@.I I INOO O! %!) $      r N)__doc__tensorboard.compat.protorr rr*r7rr r r:sf"!......''')))$# # # L11111r