+
    PjH                        ^ RI HtHtHt ^ RIt^RIHtHt ^RI	H
t
 Rt ! R R]]P                  P                  4      t ! R R	]]P                  P                  4      t ! R
 R]
]P                  P                  4      tR# )    )DictOptionalcastN)RearrangeMixinReduceMixin)_EinmixMixinzTianhe Ren & Depeng Liangc                   &   a  ] tR t^t o R tRtV tR# )	Rearrangec                $    V P                  V4      # N_apply_recipeselfinputs   &&n/Users/ahmed/devFolder/Ultron/claude-voice/gateway/.venv/lib/python3.14/site-packages/einops/layers/oneflow.pyforwardRearrange.forward       !!%((     N__name__
__module____qualname____firstlineno__r   __static_attributes____classdictcell____classdict__s   @r   r
   r
           ) )r   r
   c                   &   a  ] tR t^t o R tRtV tR# )Reducec                $    V P                  V4      # r   r   r   s   &&r   r   Reduce.forward   r   r   r   Nr   r   s   @r   r#   r#      r!   r   r#   c                   >   a  ] tR t^t o R tV 3R lR ltR tRtV tR# )EinMixc                L   \         P                  P                  \         P                  ! V4      P	                  V) V4      R R7      V n        VeN   \         P                  P                  \         P                  ! V4      P	                  V) V4      R R7      V n        R# RV n        R# )T)requires_gradN)flownn	Parameterzerosuniform_weightbias)r   weight_shapeweight_bound
bias_shape
bias_bounds   &&&&&r   _create_parametersEinMix._create_parameters   s    gg''JJ|$--|m\JZ^ ( 
 !))$**Z*@*I*I:+Wa*brv)wDIDIr   c          	      r   < V ^8  d   QhRS[ S[,          RS[ S[,          RS[ S[,          RS[ S[,          /# )   pre_reshape_patternpre_reshape_lengthspost_reshape_patternpost_reshape_lengths)r   strr   )formatr    s   "r   __annotate__EinMix.__annotate__   sI     f f%c]f &d^f 'sm	f
 'tnfr   c                    R V n         Ve    \        V3/ \        \        V4      B V n         R V n        Ve"   \        V3/ \        \        V4      B V n        R # R # r   )pre_rearranger
   r   dictpost_rearrange)r   r9   r:   r;   r<   s   &&&&&r   _create_rearrange_layersEinMix._create_rearrange_layers   s[     "*!*+>!b$tM`Ba!bD"+"+,@"eDOcDd"eD ,r   c                   V P                   e   V P                  V4      p\        P                  ! V P                  WP                  4      pV P
                  e   W P
                  ,          pV P                  e   V P                  V4      pV# r   )rB   r*   einsumeinsum_patternr/   r0   rD   )r   r   results   && r   r   EinMix.forward.   so    )&&u-ET00%E99 iiF*((0Fr   )r0   rD   rB   r/   N)	r   r   r   r   r5   rE   r   r   r   r   s   @r   r'   r'      s      f f r   r'   )typingr   r   r   oneflowr*    r   r   _einmixr   
__author__r+   Moduler
   r#   r'   r   r   r   <module>rR      s[    ' '  ) !(
) )
)[$''.. )
!\477>> !r   