+
    9iV                     h    ^ RI Ht ^ RIt^ RIHt ^ RIHtHt ^ RI	H
t
 ]! ]4      t ! R R]4      tR# )    )	getLoggerN)Fusion)TensorProtohelper	OnnxModelc                   J   a a ] tR t^t oV3R lV 3R lltR tR tRtVtV ;t	# )FusionReshapec                    < V ^8  d   QhRS[ /# )   modelr   )format__classdict__s   "e/var/www/html/photoedit/myenv/lib/python3.14/site-packages/onnxruntime/transformers/fusion_reshape.py__annotate__FusionReshape.__annotate__   s     ' 'i '    c                8   < \         SV `  VR R 4       RV n        R# )ReshapeFN)super__init__prune_graph)selfr   	__class__s   &&r   r   FusionReshape.__init__   s    	95!&r   c                F   \         P                  ! V\         P                  R 7      pV P                  P	                  RR4      p\
        P                  ! R. V.\
        P                  ! R\        P                  VP                  \        V4      RR7      R7      pWRP                  ^&   V P                  P	                  RR4      Vn        V P                  P                  V.4       V P                   P#                  V4       V P$                  V P&                  VP                  &   R	# )
)dtypeConstantconstant_shapeconst_tensorT)name	data_typedimsvalsraw)inputsoutputsvaluer   Reshape_FuseN)npasarrayint64r   create_node_namer   	make_nodemake_tensorr   INT64shapebytesinputr!   nodes_to_removeextendnodes_to_addappendthis_graph_namenode_name_to_graph_name)r   r1   reshape_nodeconcat_nodeshape_valueconstant_shape_namenew_nodes   &&&&   r   replace_reshape_node"FusionReshape.replace_reshape_node   s    jjbhh7"jj99*FVW##()$$#%++ &&;'	
 !41 JJ77	>R##[M2  *6:6J6J$$X]]3r   c                	   VP                   ^,          V9  d   R# W1P                   ^,          ,          pVP                  R8w  g5   \        VP                   4      ^8  g   \        VP                   4      ^8  d   R# V P                  P	                  V. RO. ROV4      pVf   R# Vw  rgpV P                  P	                  V. RO. ROV4      p	V	f   R# V	w  rp. pV P                  P                  VP                   ^,          4      pV^ 8X  d   VP                  ^ 4       V P                  P                  VP                   ^,          4      pV^8X  d   VP                  ^ 4       \        V4      ^8w  d   R# . p. pW.p\        VP                   4      ^8X  d   V P                  P                  VP                   ^,          4      f   V P                  P	                  V. RO. R	OV4      pVf)   V P                  P	                  V. R
O. ROV4      pVf   R# V P                  P	                  V. RO. ROV4      pVf)   V P                  P	                  V. R
O. ROV4      pVf   R# VP                  VR,          VR,          .4       VP                  R4       M\        VP                   4      ^8  d   V P                  P                  VP                   ^,          4      pVf   R# \        V\        P                  4      '       d!   VP                  VP                  4       4       MVP                  V4       \        VP                   4      ^8X  d   V P                  P                  VP                   ^,          4      f   RV9   d   R# V P                  P	                  V. RO. ROV4      pVf)   V P                  P	                  V. RO. ROV4      pVf   R# VP                  VR,          .4       VP                  R4       M\        VP                   4      ^8  d   V P                  P                  VP                   ^,          4      pVf   R# \        V\        P                  4      '       d!   VP                  VP                  4       4       MVP                  V4       VP                   ^ ,          pRpV F  pVP                   ^ ,          V8w  g   K  RpK!  	  V'       g   R# V P                  WV4       RV n        R# )   NConcatTF)	UnsqueezeGatherShape)r   r   r   )rB   r   r   )rD   MulrE   rF   )r   r   r   r   )rD   rG   SqueezeSlicerF   )r   r   r   r   r   )r   r   rB   r   )r   r   rB   r   r   )rD   DivrE   rF   )   r   r   r   )rD   rK   rH   rI   rF   )rL   r   r   r   r   )r3   op_typelenr   match_parent_pathget_constant_valuer7   r5   
isinstancer*   ndarraytolistr?   r   )r   r:   input_name_to_nodesoutput_name_to_noder;   path0unsqueeze_0gather_0shape_0path1unsqueeze_1gather_1shape_1r1   gather_valuepath2path3shape_nodesconcat_value
root_inputsame_shape_input
shape_nodes   &&&&                  r   fuseFusionReshape.fuse*   s/   a (;;)*<*<Q*?@(*c+2C2C.Dq.HCP[PaPaLbefLf

,,,	
 =+0(

,,,	
 =+0(zz44X^^A5FG1LLOzz44X^^A5FG1LLOu:?({  !Q&4::+H+HIZIZ[\I]+^+fJJ007#	E }

44E#'	 =JJ007#	E }

44E#'	 =b	5956LL""#a'::889J9J19MNL#,

33\0023\*{  !Q&4::+H+HIZIZ[\I]+^+fU{JJ007#	E }

44E#'	 =b	{+LL""#a'::889J9J19MNL#,

33\0023\*!''*
%J"j0#(  &  !!%{C  r   )r   )
__name__
__module____qualname____firstlineno__r   r?   rf   __static_attributes____classdictcell____classcell__)r   r   s   @@r   r
   r
      s!     ' 'K*C  C r   r
   )loggingr   numpyr*   fusion_baser   onnxr   r   
onnx_modelr   rh   loggerr
    r   r   <module>rv      s.       $  	8	] F ] r   