o
    ‘e   ã                   @   s(   d dl mZ dd„ ZG dd„ deƒZdS )é   )ÚOpSet8c                    s   ‡ fdd„}|S )Nc               	      s@   | d }z
ˆ | i |¤Ž}W |S    t d |jdd … |j¡ƒ‚)Nr   z%convert failed node:{}, op_type is {}é	   )Ú	ExceptionÚformatÚnameZ
layer_type)ÚargsÚkwargsÚnodeÚres©Úfunc© úUD:\Projects\ConvertPro\env\Lib\site-packages\x2paddle/op_mapper/onnx2paddle/opset9.pyÚrun_mapping   s   üÿz'print_mapping_info.<locals>.run_mappingr   )r   r   r   r   r   Úprint_mapping_info   s   
r   c                       s   e Zd Z‡ fdd„Z‡  ZS )ÚOpSet9c                    s(   t t| ƒ ||¡ | j ddgi¡ d S )NZIsNaNzpaddle.isnan)Úsuperr   Ú__init__Zdirectly_map_opsÚupdate)ÚselfÚdecoderZpaddle_graph©Ú	__class__r   r   r   !   s   zOpSet9.__init__)Ú__name__Ú
__module__Ú__qualname__r   Ú__classcell__r   r   r   r   r       s    r   N)Zopset8r   r   r   r   r   r   r   Ú<module>   s   