o
    Xi                     @   sr   d dl Z d dlZd dlmZ dd ZeejdZeejdZeejjdZ	eej
dZ
eejd	Zeejd
ZdS )    N)	optimizerc                    s    fdd}|S )Nc                     s8   t   } | i |}t   }t d||   |S )Nz time: )timeprint)argskwtsresulttefmessage Q/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/utils/timing_utils.pytimed   s
   ztimeit.<locals>.timedr   )r   r   r   r   r
   r   timeit
   s   r   LoadSaveInferzFold ConstantszRemove UnusedOptimize)r   onnx
onnxscriptr   r   loadsaveshape_inferenceinfer_shapesinferfold_constantsremove_unused_nodesremove_unusedoptimizer   r   r   r   <module>   s   