o 2h@s4dZddlmZddZddZddZd d Zd S) zUtilities for graph plugin.) graph_pb2cC d||fS)Nz%s/%s)prefixop_namerr_/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/plugins/graph/graph_util.py_prefixed_op_names rcCr)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 z%s_%sr)r func_namerrr_prefixed_func_names r c s|jD];}|j}||t|j|_fdd|jD|jdd<|jdkr>|jdr>t|jdj j|jdj _q|j j D]}|j j }||t|j j|j _qC|j j D]}|j j }||t|j|_t|j|_q^dS)Ncsg|]}t|qSr)r).0 input_namerrr /s z-_add_with_prepended_names..PartitionedCallf)nodeaddCopyFromrnameinputopattrr funclibraryfunction signaturegradient function_name gradient_func) r graph_to_adddestination_graphrnew_nodernew_funcr new_gradientrr r_add_with_prepended_names*s>           r$cCst|dkr |dSt|dkrtSt}|djjr'|j|djt|D]\}}|jj|jjkr;tdtd|d||q+|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_defrrrmerge_graph_defsPs    r0N)__doc__tensorboard.compat.protorrr r$r0rrrrs   &