+
    9i                     h    ^ RI Ht ^ RIHt ^ RIHt ^ RIHt ^ RIH	t	 ]! ]
4      t ! R R]	4      tR# )	    )	getLogger)FusionAttentionVaeFusionOptions)
ModelProto)UnetOnnxModelc                   ^   a a ] tR t^t oRV3R lV 3R llltRV3R lR lltR tRtVtV ;t	# )	VaeOnnxModelc                ,   < V ^8  d   QhRS[ RS[RS[/# )   model	num_headshidden_size)r   int)format__classdict__s   "e/var/www/html/photoedit/myenv/lib/python3.14/site-packages/onnxruntime/transformers/onnx_model_vae.py__annotate__VaeOnnxModel.__annotate__   s'     N Nj NS N3 N    c                r   < V^ 8X  d   V^ 8X  g   V^ 8  d   W2,          ^ 8X  g   Q h\         SV `  WVR7       R# )r   )r   r   N)super__init__)selfr   r   r   	__class__s   &&&&r   r   VaeOnnxModel.__init__   s:    Q;!#3Q;KbfgKghhMr   c                .   < V ^8  d   QhRS[ R,          /# )r   optionsNr   )r   r   s   "r   r   r      s     & &1E &r   c                f    \        W P                  V P                  4      pVP                  4        R # N)r   r   r   apply)r   r   self_attention_fusions   && r   fuse_multi_head_attention&VaeOnnxModel.fuse_multi_head_attention   s%     249I9I4>> Z##%r   c                    / p. ROpV F!  pV P                  V4      p\        V4      W&   K#  	  \        P                  RV 24       V# )z(
Returns node count of fused operators.
zOptimized operators:)	Attention	GroupNormSkipGroupNormNhwcConv)get_nodes_by_op_typelenloggerinfo)r   op_countopsopnodess   &    r   get_fused_operator_statistics*VaeOnnxModel.get_fused_operator_statistics   sR     
 B--b1Eu:HL  	*8*56r    )r   r   r    )
__name__
__module____qualname____firstlineno__r   r#   r2   __static_attributes____classdictcell____classcell__)r   r   s   @@r   r
   r
      s%     N N& &
 r   r
   N)loggingr   fusion_attention_vaer   fusion_optionsr   onnxr   onnx_model_unetr   r5   r,   r
   r4   r   r   <module>rA      s,     3 (  )	8	= r   