
    h                     L    S SK Jr  S SKJr  S SKJr  S SKJr  S SKJ	r	  S r
S rg)	   )	AttrValue)GraphDef)NodeDef)TensorShapeProto)
VersionDefc                 Z    SS K nUR                  U 5      nUR                  n[        U5      $ )N    )onnxloadgraphparse)fnamer
   mgs       Q/var/www/fran/franai/venv/lib/python3.13/site-packages/tensorboardX/onnx_graph.pyload_onnx_graphr      s%    		%A	A8O    c                    / n/ nSS K nUR                  U R                  U R                  5       H  nUR	                  U5        M     U H  n[        UR                  5        [        UR                  R                  R                  R                   Vs/ s H!  n[        R                  " UR                  S9PM#     snS9nUR	                  [        UR                  R                  SS9S/ [!        UR                  R                  R"                  S9[!        US9S	.S
95        M     U R$                   H  n/ nUR&                   HN  nUR	                  SR)                  UR+                  5        V	s/ s H  n	[-        U	S   5      PM     sn	5      5        MP     SR)                  U5      R                  SS9n[        UR                  S   5        UR	                  [        UR                  S   R                  SS9UR.                  UR                  S[!        US90S
95        M     0 n
U H+  nUR0                  S-   UR                  -   XR                  '   M-     [3        U[5        SS9S9$ s  snf s  sn	f )Nr	   )size)dimutf_8)encodingVariable)type)shape)dtyper   )nameopinputattrz = r   z, 
parameters)s_   )producer)nodeversions)	itertoolschainr   outputappendprintr   r   r   tensor_typer   r   Dim	dim_valuer   encoder   	elem_typer&   	attributejoin
ListFieldsstrop_typer   r   r   )r   nodes_protonodesr(   r&   d
shapeprotor    r"   fmappings              r   r   r      s   KEU\\:4  ; dii%AEAVAVA\A\A`A`aA`A!%%1;;7A`ac
W!!7!3"		(=(=(G(GH"4	 	
	  

AKK

q||~#F~!C!I~#FGH  yy%%w%7dkk!nWQ&&&8||**	D 12	
 	  G!WWs]TYY6		  R)@AA; b $Gs   (I#
,I(N)proto.attr_value_pb2r   proto.graph_pb2r   proto.node_def_pb2r   proto.tensor_shape_pb2r   proto.versions_pb2r   r   r    r   r   <module>rC      s    + % ' 4 *'Br   