o
    iv                     @   sH   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 dd Z
dS )	   )	AttrValue)GraphDef)NodeDef)TensorShapeProto)
VersionDefc                 C   s   g }dd l m  m} || }| }|d}|d}i }|D ]}|jd }	|jd }
|
||	< q"|D ].}|jd }|jd }g }d|jd	d
}|	t
|| dt|| gdt|did q5t|tdddS )N    layersedgesidnamez
from-layerzto-layerz, utf_8)encodingop
parameters)s)r   r   inputattr   )producer)nodeversions)xml.etree.ElementTreeetreeElementTreeparsegetrootfindattribjoinencodeappendr   strr   r   r   )fnamenodesETtreerootr   r	   layers_dictlayernodeidr   edge	nodeinputnodeselfr    r-   O/home/ubuntu/.local/lib/python3.10/site-packages/tensorboardX/openvino_graph.pyload_openvino_graph   s.   








r/   N)proto.attr_value_pb2r   proto.graph_pb2r   proto.node_def_pb2r   proto.tensor_shape_pb2r   proto.versions_pb2r   r/   r-   r-   r-   r.   <module>   s    