
    IЦiZ                        S r SSKrSSKrSSKJrJrJrJrJr  SSK	J
s  Jr  SSKJ
s  Jr  SSKJ
s  Jr  SSKJr  SSKJr  SSKJrJr  S/S jrS0S jrS1S	 jr   S2SS.S\S\S\S\S\4
S jjjrS3S\S\S\S\4S jjr S3S\S\S\S\4S jjrSS.S\S\4S jjr SSS.S\S\S\S\S\4
S jjr! S4SSSS.S\S\\"\\"   4   S\\"   4S jjjr#  S3S
S .S\S\S\S\4S! jjjr$S4S" jr%S#S$.S\S%\S\4S& jjr&S5S'\RN                  S(\4S) jjr S5S'\RN                  S(\4S* jjr(S+ r)\" \\R8                  S,5        \" \\R<                  5        \" \\R>                  5        \" \ 5        \" \!5        \" \#5        \" \$5        \" \%5        \" \&5        \" \\R                   S-5        \" \(\R                   S-5        \" \)\RR                  S.5        g)6a$  
The APIs in this file are exposed as `functorch.*`. They are thin wrappers
around the torch.func.* APIs that have deprecation warnings -- we're trying
to move people to the torch.func.* equivalents.

NB: We don't use *args, **kwargs in the signatures because that changes the
documentation.
    N)AnyCallableOptionalTupleUnion)	argnums_t)	in_dims_t
out_dims_tFc                 Z    Uc  SU  3nSU  SU S3nU(       a  UR                  SS5      nU$ )Nztorch.func.z[We've integrated functorch into PyTorch. As the final step of the 
integration, `functorch.zk` is deprecated as of PyTorch 
2.0 and will be deleted in a future version of PyTorch >= 2.3. 
Please use `z` instead; see the PyTorch 2.0 release notes 
and/or the `torch.func` migration guide for more details 
https://pytorch.org/docs/main/func.migrating.html
 )replace)apinew_apireplace_newlineswarnings       Z/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/_functorch/deprecated.pyget_warningr      sP    u%##&% (i  <	=  //$+N    c                 L    [        XSS9n[        R                  " U[        SS9  g )NT)r      )
stacklevel)r   warningswarnFutureWarning)r   r   r   s      r   warn_deprecatedr   '   s    #>GMM'=Q7r   c                     U R                   nUc  [        [        U5      nUR                  c  g [	        X25      nS[
        R                  " US5      -   n[
        R                  " US5      nUR                  U-   U l        g )Nz
.. warning::

z    )__name__getattr_impl__doc__r   textwrapindent)functorch_apitorch_func_apinew_api_nameapi_namer   warning_notes         r   
setup_docsr)   ,   sp    %%H 1%(1G'(//'6*JJL??<8L*22\AMr   error
chunk_sizefuncin_dimsout_dims
randomnessreturnc                F    [        SS5        [        R                  " XX#US9$ )Nvmap
torch.vmapr+   )r   apisr3   )r-   r.   r/   r0   r,   s        r   r3   r3   :   s!     FL)99THZPPr   argnumshas_auxc                 F    [        S5        [        R                  " XU5      $ )Ngrad)r   r5   r9   r-   r6   r7   s      r   r9   r9   F   s    F99TG,,r   c                 F    [        S5        [        R                  " XU5      $ )Ngrad_and_value)r   r5   r<   r:   s      r   r<   r<   K   s      $%tg66r   )r7   c                L    [        S5        [        R                  " U /UQ7SU06$ )Nvjpr7   )r   r    r>   )r-   r7   primalss      r   r>   r>   R   s#    E99T5G5W55r   strictr7   r?   tangentsrA   c                D    [        S5        [        R                  " XX#US9$ )Njvpr@   )r   r    rD   )r-   r?   rB   rA   r7   s        r   rD   rD   W   s     E99THWMMr   r7   r,   _preallocate_and_copyr,   c                H    [        S5        [        R                  " U UUUUS9$ )NjacrevrE   )r   r    rH   )r-   r6   r7   r,   rF   s        r   rH   rH   c   s,     H<<3 r   r0   c                B    [        S5        [        R                  " XX#S9$ )NjacfwdrI   )r   r    rK   )r-   r6   r7   r0   s       r   rK   rK   u   s     H<<wFFr   c                 @    [        S5        [        R                  " XS9$ )Nhessian)r6   )r   r    rM   )r-   r6   s     r   rM   rM      s    I==//r   	mutationsremoverP   c                @    [        S5        [        R                  " XS9$ )NfunctionalizerO   )r   r    rR   )r-   rP   s     r   rR   rR      s    O$t33r   modeldisable_autograd_trackingc                 F    [        SS5        [        R                  " X5      $ )Nmake_functionaltorch.func.functional_call)r   _nn_implrV   rS   rT   s     r   rV   rV      s    %'CD##EEEr   c                 F    [        SS5        [        R                  " X5      $ )Nmake_functional_with_buffersrW   )r   rX   r[   rY   s     r   r[   r[      s!     24PQ00RRr   c                 F    [        SS5        [        R                  " U 5      $ )Ncombine_state_for_ensembletorch.func.stack_module_state)r   rX   r]   )modelss    r   r]   r]      s    02QR..v66r   r4   rW   r^   )NF)N)NN)r   r   r*   )r   F)r   )F)*r!   r"   r   typingr   r   r   r   r   torch._functorch.apis
_functorchr5   !torch._functorch.eager_transformseager_transformsr     torch._functorch.make_functionalrV   rX   torch.nnnnr   torch._functorch.vmapr	   r
   r   r   r)   strr3   boolr9   r<   r>   rD   intrH   rK   rM   rR   Moduler[   r]    r   r   <module>rn      s     8 8 $ $ 1 1 3 3  7 7 8
B  		Q 	Q
	Q	Q 	Q 		Q 	Q-x -) -$ -8 - =B7
7&75977 38 6h 64 6 	N
	N	N 	N
 	N 	N '(  $
3c
?#
 ( G
 G
GG G
 G0
 4? 4 4S 48 4
F299 F F 9>S99S15S7
 4L ) 
4  
>4.. / 
3 
3 
6  
6  
7  
=  
?H446R S 
 (":":<X ''#r   