§
    Ì²Åj:  ã                  óP   — d dl mZ d dlmZmZ dd„Z ed¦  «        Zdd
„Zdd„Z	dS )é    )Úannotations)ÚNativeFunctionsGroupÚNativeFunctionsViewGroupÚgú/NativeFunctionsGroup | NativeFunctionsViewGroupÚreturnÚstrc                ó®   — t          | t          ¦  «        r(t          | j        j        j        j        j        ¦  «        S t          | j        j        ¦  «        S ©N)	Ú
isinstancer   r	   Ú
functionalÚfuncÚnameÚbaseÚviewÚ	root_name)r   s    úOC:\Projects\Historical\venv\Lib\site-packages\torchgen/static_runtime/config.pyÚfunc_name_base_strr      sD   € Ý�!Õ)Ñ*Ô*ð %Ý�1”<Ô$Ô)Ô.Ô3Ñ4Ô4Ð4å�1”6Ô#Ñ$Ô$Ð$ó    )'ÚabsÚaddÚaddmmÚallÚanyÚargminÚbmmÚclampÚ	clamp_minÚcumsumÚdivÚfmodÚindex_selectÚ
leaky_reluÚlinearÚlogÚmatmulÚmulÚnarrow_copyÚnonzeroÚpowÚ	remainderÚsigmoidÚsignÚsubÚtanhÚdetachÚ	expand_asÚflattenÚnarrowÚ
reshape_asÚselectÚsliceÚsoftmaxÚsplitÚsqueezeÚ	transposer   ÚwhereÚboolc                ó2   — t          | ¦  «        }|t          v S r   )r   Úis_hand_written_ops_)r   Ú	name_bases     r   Úis_hand_writtenr@   :   s   € Ý" 1Ñ%Ô%€IØÕ,Ð,Ð,r   Úarg_mapúdict[str, str]Úop_nameÚindexÚintÚNonec                ó  — |dk    s|dk    sJ ‚|dk    r'|dk    rd| d<   d| d<   d| d<   nd	| d<   d
| d<   d
| d<   d S |dk    r|dk    rd| d<   d| d<   n
d	| d<   d
| d<   d S |dk    r|dk    rd| d<   nd	| d<   d S |dk    r|dk    rd| d<   d| d<   n
d| d<   d| d<   d S |dk    r|dk    rd| d<   nd| d<   d S |dk    r|dk    rd| d<   nd| d<   d S |dk    r|dk    rd| d<   nd| d<   d S |dk    r|dk    rd| d<   nd	| d<   d S |d k    r|dk    rd| d<   nd	| d<   d S |d!k    r|dk    rd| d"<   d#| d$<   n
d
| d"<   d#| d$<   d S |d%k    r|dk    rd| d"<   d#| d$<   n
d
| d"<   d#| d$<   d S |d&k    r'|dk    rd| d<   d'| d(<   d| d)<   nd	| d<   d*| d(<   d
| d)<   d S |d+k    r|dk    rd| d<   d,| d(<   n
d	| d<   d-| d(<   d S |d.k    r'|dk    rd| d<   d'| d(<   d| d)<   nd	| d<   d*| d(<   d
| d)<   d S |d/k    r'|dk    rd0| d<   d1| d(<   d| d)<   nd2| d<   d3| d(<   d
| d)<   d S |d4v rd5| d6<   d S |d7k    r|dk    rd0| d<   d8| d9<   n
d2| d<   d8| d9<   d S |d:k    r'|dk    rd;| d<   d<| d=<   d;| d<   nd>| d<   d?| d=<   d>| d<   d S |d@k    r|dk    rdA| d<   ndB| d<   d S |dCk    r'|dk    rdD| dE<   dD| d<   dF| d<   ndG| dE<   dG| d<   dH| d<   d S |dIk    r'|dk    rdJ| dE<   dJ| d<   dK| d<   ndL| dE<   dL| d<   dM| d<   d S |dNk    r|dk    rdO| d<   dP| d<   n
dQ| d<   dR| d<   d S |dSk    r|dk    rdT| d<   dU| d<   n
dV| d<   dW| d<   d S |dXk    r1|dk    rdY| d<   dZ| d[<   d\| d<   d]| d^<   nd_| d<   dZ| d[<   d`| d<   d]| d^<   d S |dak    r|dk    rdb| d<   dc| dd<   n
de| d<   dc| dd<   d S |dfk    r'|dk    rdb| dE<   db| d<   dc| dd<   nde| dE<   de| d<   dc| dd<   d S |dgk    r;|dk    rd;| d<   dh| d[<   di| d<   d;| dj<   d8| dk<   ndl| d<   dh| d[<   dm| d<   dl| dj<   d8| dk<   d S |dnk    r1|dk    rd;| d<   dh| d[<   do| d<   d;| dj<   ndp| d<   dh| d[<   dq| d<   dp| dj<   d S |drk    r'|dk    rds| d<   ds| d<   dZ| d[<   nd| d<   d| d<   dZ| d[<   d S |dtk    rO|dk    r$du| dE<   d| d<   dv| d(<   d| d)<   dZ| dw<   dZ| dx<   du| dy<   n#du| dE<   dz| d<   d{| d(<   dz| d)<   dZ| dw<   dZ| dx<   du| dy<   d S |d|v r8|dk    rd}| d<   d\| d<   d}| d~<   nd| d<   d€| d<   d| d~<   d�| v r|d‚k    rdƒnd„| d�<   d S |d…k    rd†| d�<   |dk    rd1| d<   nd3| d<   d S |d‡k    r|dk    rdˆ| d<   dˆ| d<   n
d‰| d<   d‰| d<   d S |dŠk    r'|dk    rd‹| dŒ<   d�| dŽ<   d]| d�<   nd�| dŒ<   d‘| dŽ<   d]| d�<   d S |d’k    r'|dk    rd“| d<   d”| d•<   d]| d�<   nd–| d<   d—| d•<   d]| d�<   d S |d˜v rdh| d™<   d8| dš<   dZ| d›<   d S d S )œNr   é   Úaddrzat::rand({6, 6})Úselfzat::rand({6})Úvec1Úvec2zat::rand({22, 22})zat::rand({22})ÚmvÚvecÚaddbmmÚcrosszat::rand({3, 3, 3})Úotherzat::rand({22, 3, 22})Útakez(at::randint(0, 216, {20}, torch::kInt64)rD   z*at::randint(0, 1000, {100}, torch::kInt64)Útake_along_dimzat::argsort(self0, 1, true)Úindiceszat::argsort(self1, 1, true)Úmasked_selectzat::randn({6, 6, 6}) > 0.5Úmaskzat::rand({22, 22, 22}) > 0.5ÚorgqrÚinput2ÚormqrÚquantileÚqz"linear"ÚinterpolationÚnanquantileÚmulti_margin_lossz"at::randint(6, {6}, torch::kInt64)ÚtargetÚweightz$at::randint(22, {22}, torch::kInt64)Úmultilabel_margin_lossz%at::randint(6, {6, 6}, torch::kInt64)z(at::randint(22, {22, 22}, torch::kInt64)Únll_lossÚ
nll_loss2dzat::rand({6, 6, 6, 6})z(at::randint(6, {6, 6, 6}, torch::kInt64)zat::rand({22, 22, 22, 22})z,at::randint(22, {22, 22, 22}, torch::kInt64))Úfft_fftÚfft_ifftÚfft_rfftÚ	fft_irfftÚfft_hfftÚ	fft_ihfftz	"forward"ÚnormÚlinalg_tensorinvÚ2ÚindÚaddmvzat::rand({2})zat::rand({2, 2})Úmatzat::rand({35})zat::rand({35, 35})Úacoshz)at::rand({2, 2, 2}) + at::ones({2, 2, 2})z)at::rand({5, 5, 5}) + at::ones({5, 5, 5})Úadaptive_max_pool2d_backwardzat::rand({2, 2, 2}, at::kFloat)Úgrad_outputz'at::randint(0, 1, {2, 2, 2}, at::kLong)zat::rand({3, 3, 3}, at::kFloat)z'at::randint(0, 1, {3, 3, 3}, at::kLong)Úadaptive_max_pool3d_backwardz"at::rand({2, 2, 2, 2}, at::kFloat)z*at::randint(0, 1, {2, 2, 2, 2}, at::kLong)z"at::rand({3, 3, 3, 3}, at::kFloat)z*at::randint(0, 1, {3, 3, 3, 3}, at::kLong)Úbitwise_left_shiftz+at::randint(1, 1 << 4, {6, 6, 6}, at::kInt)z'at::randint(1, 26, {6, 6, 6}, at::kInt)z.at::randint(1, 1 << 4, {22, 22, 22}, at::kInt)z*at::randint(1, 26, {22, 22, 22}, at::kInt)Úbitwise_right_shiftz2at::randint(1 << 21, 1 << 30, {6, 6, 6}, at::kInt)z'at::randint(1, 22, {6, 6, 6}, at::kInt)z5at::randint(1 << 21, 1 << 30, {22, 22, 22}, at::kInt)z*at::randint(1, 22, {22, 22, 22}, at::kInt)Úgatherz&at::randint(1, 100, {2,2,2}, at::kInt)Ú1Údimz)at::randint(0, 1, {2,2,2}, torch::kInt64)ÚfalseÚsparse_gradz&at::randint(1, 100, {5,5,5}, at::kInt)z)at::randint(0, 4, {5,5,5}, torch::kInt64)Úgeluzat::rand({6, 6, 6})z"tanh"Úapproximatezat::rand({22, 22, 22})Úgelu_backwardÚ	index_addÚ0z at::randint(0, 1, {2}, at::kInt)ÚsourceÚalphazat::rand({16})z"at::randint(0, 10, {16}, at::kInt)Ú
index_copyz!at::randint(0, 1, {2}, at::kLong)zat::rand({32})z#at::randint(0, 10, {32}, at::kLong)Úlinalg_crosszat::rand({6, 3, 6})Únll_loss_backwardzat::rand({})z%at::randint(0, 5, {6}, torch::kInt64)Ú	reductionÚignore_indexÚtotal_weightzat::rand({36})z'at::randint(0, 11, {36}, torch::kInt64))ÚscatterÚscatter_addÚ_scatter_reducez+at::randint(1, 100, {2,2,2}, torch::kInt64)Úsrcz+at::randint(1, 100, {5,5,5}, torch::kInt64)z)at::randint(0, 1, {5,5,5}, torch::kInt64)ÚreducerŠ   z"sum"z"add"Úscatter_reducez"mean"Úspecial_zetaz2at::rand({2,2,2}, at::kDouble) + at::ones({2,2,2})z2at::rand({5,5,5}, at::kDouble) + at::ones({5,5,5})Ú _convert_indices_from_csr_to_cooz!torch::tensor({1}, torch::kInt32)Úcrow_indicesz'torch::tensor({0, 1, 0}, torch::kInt32)Úcol_indicesÚ	out_int32z!torch::tensor({0}, torch::kInt32)zBtorch::tensor({0, 1, 0, 2, 1, 2, 0, 1, 0, 2, 1, 2}, torch::kInt32)Ú _convert_indices_from_coo_to_csrz at::randint(0, 3, {2}, at::kInt)Ú10Úsizez!at::randint(0, 3, {12}, at::kInt)Ú24)ÚdiagonalÚlinalg_diagonalÚoffsetÚdim1Údim2© )rA   rC   rD   s      r   Úoverride_test_valuesr�   ?   sœ
  € Ø�AŠ:ˆ:˜ !š˜˜Ð#Ø�&ÒÐØ�AŠ:ˆ:Ø0ˆG�F‰OØ-ˆG�F‰OØ-ˆG�F‰OˆOà2ˆG�F‰OØ.ˆG�F‰OØ.ˆG�F‰OØˆØ�$‚€Ø�AŠ:ˆ:Ø0ˆG�F‰OØ,ˆG�E‰NˆNà2ˆG�F‰OØ-ˆG�E‰NØˆØ�(ÒÐØ�AŠ:ˆ:Ø0ˆG�F‰OˆOà2ˆG�F‰OØˆØ�'ÒÐØ�AŠ:ˆ:Ø3ˆG�F‰OØ4ˆG�GÑÐà5ˆG�F‰OØ6ˆG�GÑØˆØ�&ÒÐØ�AŠ:ˆ:ØIˆG�GÑÐàKˆG�GÑØˆØÐ"Ò"Ð"Ø�AŠ:ˆ:Ø!>ˆG�IÑÐà!>ˆG�IÑØˆØ�/Ò!Ð!Ø�AŠ:ˆ:Ø:ˆG�F‰OˆOà<ˆG�F‰OØˆØ�'ÒÐØ�AŠ:ˆ:Ø 2ˆG�HÑÐà 4ˆG�HÑØˆØ�'ÒÐØ�AŠ:ˆ:Ø 2ˆG�HÑÐà 4ˆG�HÑØˆØ�*ÒÐØ�AŠ:ˆ:Ø*ˆG�C‰LØ'1ˆG�OÑ$Ð$à+ˆG�C‰LØ'1ˆG�OÑ$ØˆØ�-ÒÐØ�AŠ:ˆ:Ø*ˆG�C‰LØ'1ˆG�OÑ$Ð$à+ˆG�C‰LØ'1ˆG�OÑ$ØˆØÐ%Ò%Ð%Ø�AŠ:ˆ:Ø0ˆG�F‰OØ DˆG�HÑØ /ˆG�HÑÐà2ˆG�F‰OØ FˆG�HÑØ 0ˆG�HÑØˆØÐ*Ò*Ð*Ø�AŠ:ˆ:Ø0ˆG�F‰OØ GˆG�HÑÐà2ˆG�F‰OØ JˆG�HÑØˆØ�*ÒÐØ�AŠ:ˆ:Ø0ˆG�F‰OØ DˆG�HÑØ /ˆG�HÑÐà2ˆG�F‰OØ FˆG�HÑØ 0ˆG�HÑØˆØ�,ÒÐØ�AŠ:ˆ:Ø6ˆG�F‰OØ JˆG�HÑØ /ˆG�HÑÐà:ˆG�F‰OØ NˆG�HÑØ 0ˆG�HÑØˆØð ð ð ð &ˆ�‰ØˆØÐ$Ò$Ð$Ø�AŠ:ˆ:Ø6ˆG�F‰OØ ˆG�E‰NˆNà:ˆG�F‰OØ ˆG�E‰NØˆØ�'ÒÐØ�AŠ:ˆ:Ø-ˆG�F‰OØ/ˆG�E‰NØ,ˆG�E‰NˆNà.ˆG�F‰OØ1ˆG�E‰NØ-ˆG�E‰NØˆØ�'ÒÐØ�AŠ:ˆ:ØIˆG�F‰OˆOàIˆG�F‰OØˆØÐ0Ò0Ð0Ø�AŠ:ˆ:Ø%FˆG�MÑ"Ø?ˆG�F‰OØ!JˆG�IÑÐà%FˆG�MÑ"Ø?ˆG�F‰OØ!JˆG�IÑØˆØÐ0Ò0Ð0Ø�AŠ:ˆ:Ø%IˆG�MÑ"ØBˆG�F‰OØ!MˆG�IÑÐà%IˆG�MÑ"ØBˆG�F‰OØ!MˆG�IÑØˆØÐ&Ò&Ð&Ø�AŠ:ˆ:ØKˆG�F‰OØHˆG�GÑÐàNˆG�F‰OØKˆG�GÑØˆØÐ'Ò'Ð'Ø�AŠ:ˆ:ØRˆG�F‰OØHˆG�GÑÐàUˆG�F‰OØKˆG�GÑØˆØ�(ÒÐØ�AŠ:ˆ:ØFˆG�F‰OØ ˆG�E‰NØJˆG�GÑØ%,ˆG�MÑ"Ð"àFˆG�F‰OØ ˆG�E‰NØJˆG�GÑØ%,ˆG�MÑ"ØˆØ�&ÒÐØ�AŠ:ˆ:Ø3ˆG�F‰OØ%-ˆG�MÑ"Ð"à6ˆG�F‰OØ%-ˆG�MÑ"ØˆØ�/Ò!Ð!Ø�AŠ:ˆ:Ø%:ˆG�MÑ"Ø3ˆG�F‰OØ%-ˆG�MÑ"Ð"à%=ˆG�MÑ"Ø6ˆG�F‰OØ%-ˆG�MÑ"ØˆØ�+ÒÐØ�AŠ:ˆ:Ø-ˆG�F‰OØ ˆG�E‰NØAˆG�GÑØ /ˆG�HÑØ"ˆG�GÑÐà.ˆG�F‰OØ ˆG�E‰NØCˆG�GÑØ 0ˆG�HÑØ"ˆG�GÑØˆØ�,ÒÐØ�AŠ:ˆ:Ø-ˆG�F‰OØ ˆG�E‰NØBˆG�GÑØ /ˆG�HÑÐà.ˆG�F‰OØ ˆG�E‰NØDˆG�GÑØ 0ˆG�HÑØˆØ�.Ò Ð Ø�AŠ:ˆ:Ø3ˆG�F‰OØ4ˆG�GÑØ ˆG�E‰NˆNà5ˆG�F‰OØ6ˆG�GÑØ ˆG�E‰NØˆØÐ%Ò%Ð%Ø�AŠ:ˆ:Ø%3ˆG�MÑ"Ø-ˆG�F‰OØ GˆG�HÑØ /ˆG�HÑØ#&ˆG�KÑ Ø&)ˆG�NÑ#Ø&4ˆG�NÑ#Ð#à%3ˆG�MÑ"Ø.ˆG�F‰OØ IˆG�HÑØ 0ˆG�HÑØ#&ˆG�KÑ Ø&)ˆG�NÑ#Ø&4ˆG�NÑ#ØˆØÐ?Ð?Ð?Ø�AŠ:ˆ:ØKˆG�F‰OØJˆG�GÑØJˆG�E‰NˆNàKˆG�F‰OØJˆG�GÑØJˆG�E‰NØ�wÐÐØ+2Ð6GÒ+GÐ+G  ÈWˆG�HÑØˆØÐ"Ò"Ð"Ø$ˆ�ÑØ�AŠ:ˆ:ØIˆG�GÑÐàMˆG�GÑØˆØ�.Ò Ð Ø�AŠ:ˆ:ØRˆG�F‰OØSˆG�GÑÐàRˆG�F‰OØSˆG�GÑØˆØÐ4Ò4Ð4Ø�AŠ:ˆ:Ø&IˆG�NÑ#Ø%NˆG�MÑ"Ø#*ˆG�KÑ Ð à&IˆG�NÑ#àTð �MÑ"ð $+ˆG�KÑ ØˆØÐ4Ò4Ð4Ø�AŠ:ˆ:Ø@ˆG�F‰OØ"ˆG�F‰OØ#*ˆG�KÑ Ð àAˆG�F‰OØ"ˆG�F‰OØ#*ˆG�KÑ ØˆØÐ1Ð1Ð1Øˆ�ÑØˆ�‰Øˆ�‰Øˆð	 2Ð1r   N)r   r   r   r	   )r   r   r   r<   )rA   rB   rC   r	   rD   rE   r   rF   )
Ú
__future__r   Útorchgen.modelr   r   r   Ú	frozensetr>   r@   r�   rœ   r   r   ú<module>r¡      s•   ðØ "Ð "Ð "Ð "Ð "Ð "à IÐ IÐ IÐ IÐ IÐ IÐ IÐ Ið%ð %ð %ð %ð !�yð(ñ*ô *Ð ðZ-ð -ð -ð -ð
Eð Eð Eð Eð Eð Er   