
    h              
           S SK JrJrJr  SSKJr  SSKJr  SSKJ	r	J
r
  SSKJr  \
r\	r\R                  " S5      S\\\4   4S	 j5       rS
\\\4   S\S\S\\\4   4S jrg)    )CallableTuplecast   )registry)Model)Floats2dRagged)	ArrayInfozreduce_max.v1returnc                  "    [        S[        5      $ )N
reduce_max)r   forward     Q/home/james-whalen/.local/lib/python3.13/site-packages/thinc/layers/reduce_max.pyr   r      s    w''r   modelXris_trainc                   ^ ^^^ T R                   R                  [        [        UR                  5      UR
                  5      u  nmUR
                  m[        R                  " U5      mS[        S[        4UUU U4S jjnX44$ )NdYr   c                 t   > TR                  U 5        [        TR                  R                  U TT5      T5      $ )N)check_consistencyr
   opsbackprop_reduce_max)r   
array_infolengthsr   whichs    r   backpropforward.<locals>.backprop   s1    $$R(eii33BwGQQr   )
r   r   r   r	   datar   r   
from_arrayOutTInT)r   r   r   Yr   r   r   r   s   `    @@@r   r   r      sk    yy##D277$;RZZHHAujjG%%a(JRT Rc R R ;r   N)typingr   r   r   configr   r   r   typesr	   r
   utilr   r$   r#   layersr   boolr   r   r   r   <module>r,      s    ( (   $  
!(E#t)$ ( "(	5d# 	 	 	tX~AV 	r   