
    [Th+                         S SK JrJr  S SKJrJr  S SKJrJr  S SK	J
r
  SSKJr  \" \SS9S	\S
\\
/\4   S\4S j5       r\" \SS9S	\S
\\/\4   S\4S j5       rSS/rg)    )AnyCallable)_fx_map_aggregate_fx_map_arg)immutable_dictimmutable_list)Node   )substitute_in_graphT)can_constant_fold_throughafnreturnc                 $   ^ [        U U4S j5      $ )Nc                 B   > [        U [        5      (       a  T" U 5      $ U $ N)
isinstancer	   )xr   s    R/var/www/auris/envauris/lib/python3.13/site-packages/torch/_dynamo/polyfills/fx.py<lambda>map_arg.<locals>.<lambda>   s    z!T/B/Bbe&I&I    map_aggregate)r   r   s    `r   map_argr   
   s    IJJr   c                   ^ [        U [        5      (       a:  U4S jU  5       n[        U S5      (       a  [        U 5      " U6 nU$ [        U5      nU$ [        U [        5      (       a(  [        U  Vs/ s H  n[        UT5      PM     sn5      nU$ [        U [        5      (       a<  [        U R                  5        VVs/ s H  u  pVU[        UT5      4PM     snn5      nU$ [        U [        5      (       aK  [        [        U R                  T5      [        U R                  T5      [        U R                  T5      5      nU$ T" U 5      nU$ s  snf s  snnf )Nc              3   <   >#    U  H  n[        UT5      v   M     g 7fr   r   ).0elemr   s     r   	<genexpr> map_aggregate.<locals>.<genexpr>   s     4!$mD"%%!s   _fields)r   tuplehasattrtypelistr   r   dictr   itemsslicestartstopstep)r   r   itresultr   kvs    `     r   r   r      s+    !U4!4!(I!6!6a" M =B"I M 
At		Q GQTtR!8Q GH M 
At		qwwy Qytq!]1b%9!:y QR M 
Au		!''2&!&&"%!&&"%
 M AM !H Qs   /E	;E
r   r   N)typingr   r   torch._Cr   r   torch.fx.immutable_collectionsr   r   torch.fx.noder	   
decoratorsr   r   r   __all__ r   r   <module>r8      s      3 I  , [DAKs K$- K# K BK &$GS hucz2 s  H, r   