o
    Xi                     @  sb   d Z ddlmZ ddlmZ ddlZddlmZ ddlmZ	 ddl
mZ dd	gZ	ddddZdS )z;Deprecated. This module is kept for backward compatibility.    )annotations)SequenceN)pattern)rewrite)ORT_PATTERN_REWRITE_RULESr   r   pattern_rules$Sequence[pattern.RewriteRule] | Nonemodel_protoonnx.ModelProtoreturnc                C  s   |pt }t| |dS )a
  Rewrite the model using the given rules.

    Args:
        model_proto: The model to rewrite.
        pattern_rules: The pattern rewrite rules to apply. If None, the default rules
            for onnxruntime are used.

    Returns:
        The rewritten model.
    )pattern_rewrite_rules)r   _rewrite)r	   r    r   \/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/rewriter/onnxruntime/__init__.pyr      s   )N)r   r   r	   r
   r   r
   )__doc__
__future__r   typingr   onnxonnxscript.rewriterr   r   r   onnxscript.rewriter.ort_fusionsr   __all__r   r   r   r   <module>   s   	