o
    e~                     @   s4   d dl Z d dlmZmZmZ dddZdddZdS )	    N)loadchecker	NodeProtoreturnc                  C   s>   t d} | jdt dd |  }t|j}t| d S )Nzcheck-modelmodel_pbrbtype)	argparseArgumentParseradd_argumentFileType
parse_argsr   r   r   check_model)parserargsmodel r   @D:\Projects\ConvertPro\env\Lib\site-packages\onnx/bin/checker.pyr      s
   

r   c                  C   sJ   t d} | jdt dd |  }t }||j  t	
| d S )Nz
check-nodenode_pbr   r   )r
   r   r   r   r   r   ZParseFromStringr   readr   
check_node)r   r   noder   r   r   r      s   
r   )r   N)r
   Zonnxr   r   r   r   r   r   r   r   r   <module>   s   
	