o Xi$@sdZddlmZgdZddlZddlZddlZe e Z Gdddej j ZGdddej j ZGd d d ej j ZGd d d ej j ZdS) zLift constants to initializers.) annotations)AddInitializersToInputsPassLiftConstantsToInitializersPass'LiftSubgraphInitializersToMainGraphPass RemoveInitializersFromInputsPassNcs8eZdZdZddfdd ZdddZdddZZS)raLift constants to initializers. Attributes: lift_all_constants: Whether to lift all Constant nodes, including those that does not contain a tensor attribute (e.g. with value_ints etc.) Default to False, where only Constants with the ``value`` attribute are lifted. size_limit: The minimum size of the tensor to be lifted. If the tensor contains number of elements less than size_limit, it will not be lifted. Default is 16. Flift_all_constantsbool size_limitintcst||_||_dS)N)super__init__rr )selfrr  __class___/home/ubuntu/.local/lib/python3.10/site-packages/onnx_ir/passes/common/constant_manipulation.pyr !s  z(LiftConstantsToInitializersPass.__init__modelir.Modelreturnir.passes.PassResultc Csd}tj|jD]}|jdusJ|jdks|jdvrq |jdr,t d|j q t |j }t|dkrAt d|j q tt|j \}}|jdj }|dusXJt|tjs`J|||||}|durmq tj||jt|j||jdjd} | j|jdj|jdusJ|j| |jd| |jj|dd |d7}t d |j |q |rt d |tj j!|t"|d S) NrConstant)zonnx.aizs  t6