o
    Xi                     @   s   d dl mZmZ dd Zdd Zdd Zdd	 Zd
d Zdd Zdd Z	eee	Z
eee	Zeee	Zeee	Zeee	Zeee	Zeg e
 e eeeeZdS )    )RewriteRuleRewriteRuleSetc                 C   s   |d S N    opxr   r   [/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/rewriter/rules/common/_no_op.pymul_by_1
      r   c                 C   s   |d S Nr   r   r   r   r   r
   add_0   r   r   c                 C   s   |d S r   r   r   r   r   r
   sub_0   r   r   c                 C   s   |d S r   r   r   r   r   r
   div_by_1   r   r   c                 C      | j |ddS )Ng        )ratioDropoutr   r   r   r
   dropout_zero      r   c                 C   r   )NF)training_moder   r   r   r   r
   dropout_inference   r   r   c                 K   s
   |  |S )N)Identity)r   r	   _r   r   r
   identity#   s   
r   N)!onnxscript.rewriter._rewrite_ruler   r   r   r   r   r   r   r   r   mul_by_1_rule
add_0_rule
sub_0_rulediv_by_1_ruledropout_zero_ruledropout_inference_rulecommuterulesr   r   r   r
   <module>   s:   





