§
    S�’jn  ã                   ój   — d dl T d dlZd dlmZ dgZdededefd„Zdededefd	„Z G d
„ d¦  «        Z	dS )é    )Ú*N)ÚConstantIntNodeÚNestedIntNodeÚlhsÚrhsÚreturnc                 ó–   — t          | t          ¦  «        o4t          |t          ¦  «        o| j        |j        k    o| j        |j        k    S ©N)Ú
isinstancer   Út_idÚcoeff©r   r   s     úOC:\Projects\DSCons\.venv\Lib\site-packages\torch\nested\_internal\nested_int.pyÚ_eqr      sJ   € å�3�Ñ&Ô&ð 	#Ý�s�MÑ*Ô*ð	#àŒH˜œÒ ð	#ð ŒI˜œÒ"ð	ó    c                 ó  — t          |t          ¦  «        rDt          | t          ¦  «        r/| j        |j        k    r| j        |j        k    S t	          d¦  «        ‚t          | t          ¦  «        r=|                     ¦   «         r|                     ¦   «         dk    rdS t	          d¦  «        ‚t          |t          ¦  «        r=|                      ¦   «         r|                      ¦   «         dk     rdS t	          d¦  «        ‚t	          d¦  «        ‚)Nzge: relation is indeterminateé   TFzinputs unsupported)r   r   r   r   Ú
ValueErrorÚis_constantÚconstant_intr   s     r   Ú_ger      sý   € Ý�#•}Ñ%Ô%ð /­*°S½-Ñ*HÔ*Hð /ØŒ8�s”xÒÐØ”9 ¤	Ò)Ð)ÝÐ8Ñ9Ô9Ð9Ý	�C�Ñ	'Ô	'ð 	/Ø�?Š?ÑÔð 	 ×!1Ò!1Ñ!3Ô!3°qÒ!8Ð!8Ø�4ÝÐ8Ñ9Ô9Ð9Ý	�C�Ñ	'Ô	'ð /Ø�?Š?ÑÔð 	 ×!1Ò!1Ñ!3Ô!3°aÒ!7Ð!7Ø�5ÝÐ8Ñ9Ô9Ð9åÐ-Ñ.Ô.Ð.r   c                   ó^  — e Zd Zdededdfd„Zdefd„Zdee         fd„Zdefd„Z	defd	„Z
defd
„Zdefd„Zdd„Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Zdedd fd„Zdedefd„Zdedefd„Zdedefd„Zdedefd„Zdedefd„Zdedefd„Zdefd„Zdefd„Zdefd„Zdedefd„ZdS ) r   r   r   r   Nc                 ó"   — || _         || _        d S r
   )r   r   )Úselfr   r   s      r   Ú__init__zNestedIntNode.__init__&   s   € ØˆŒ	ØˆŒ
ˆ
ˆ
r   c                 ó   — | j         S r
   )r   ©r   s    r   Únested_int_coeffzNestedIntNode.nested_int_coeff*   s
   € ØŒzÐr   c                 ó   — d S r
   © r   s    r   Úmaybe_as_intzNestedIntNode.maybe_as_int-   ó   € Øˆtr   c                 ó   — dS ©NTr    r   s    r   Úis_intzNestedIntNode.is_int0   r"   r   c                 ó   — dS ©NFr    r   s    r   Úis_floatzNestedIntNode.is_float3   ó   € Øˆur   c                 ó   — dS r'   r    r   s    r   Úis_boolzNestedIntNode.is_bool6   r)   r   c                 ó   — dS r$   r    r   s    r   Úis_nested_intzNestedIntNode.is_nested_int9   r"   r   c                 ó   — | S r
   r    r   s    r   ÚclonezNestedIntNode.clone<   s   € Øˆr   c                 óN   — | j         dk    r
d| j        › �S | j         › d| j        › �S )Né   Újz*j)r   r   r   s    r   Ú_strzNestedIntNode._str?   s3   € ØŒ:˜Š?ˆ?Ø"�t”y�?�?Ð"Ø”*Ð+Ð+ ¤	Ð+Ð+Ð+r   c                 ó*   — |                       ¦   «         S r
   ©r3   r   s    r   ÚstrzNestedIntNode.strD   ó   € Ø�yŠy‰{Œ{Ðr   c                 ó*   — |                       ¦   «         S r
   r5   r   s    r   Ú__str__zNestedIntNode.__str__G   r7   r   c                 ó*   — |                       ¦   «         S r
   r5   r   s    r   Ú__repr__zNestedIntNode.__repr__J   r7   r   c                 ó*   — |                       ¦   «         S r
   r5   r   s    r   Ú_graph_reprzNestedIntNode._graph_reprM   r7   r   Úotherc                 óÌ   — |                      ¦   «         r|                     ¦   «         }nt          dt          |¦  «        › �¦  «        ‚t	          | j        | j        |z  ¦  «        S )Nzunsupported: )r   r   r   Útyper   r   r   ©r   r>   s     r   ÚmulzNestedIntNode.mulP   s_   € Ø×ÒÑÔð 	<Ø×&Ò&Ñ(Ô(ˆEˆEåÐ:­T°%©[¬[Ð:Ð:Ñ;Ô;Ð;Ý˜TœY¨¬
°UÑ(:Ñ;Ô;Ð;r   c                 ó\   — t           j                             t          | |¦  «        ¦  «        S r
   ©ÚtorchÚ_CÚ_get_constant_bool_symnoder   rA   s     r   ÚeqzNestedIntNode.eqW   ó"   € ÝŒx×2Ò2µ3°t¸UÑ3CÔ3CÑDÔDÐDr   c                 ó^   — t           j                             t          | |¦  «         ¦  «        S r
   rD   rA   s     r   ÚnezNestedIntNode.neZ   ó%   € ÝŒx×2Ò2µs¸4ÀÑ7GÔ7GÐ3GÑHÔHÐHr   c                 ó^   — t           j                             t          || ¦  «         ¦  «        S r
   ©rE   rF   rG   r   rA   s     r   ÚgtzNestedIntNode.gt]   s%   € ÝŒx×2Ò2µs¸5À$Ñ7GÔ7GÐ3GÑHÔHÐHr   c                 ó^   — t           j                             t          | |¦  «         ¦  «        S r
   rN   rA   s     r   ÚltzNestedIntNode.lt`   rL   r   c                 ó\   — t           j                             t          || ¦  «        ¦  «        S r
   rN   rA   s     r   ÚlezNestedIntNode.lec   s"   € ÝŒx×2Ò2µ3°u¸dÑ3CÔ3CÑDÔDÐDr   c                 ó\   — t           j                             t          | |¦  «        ¦  «        S r
   rN   rA   s     r   ÚgezNestedIntNode.gef   rI   r   c                 ó   — dS r'   r    r   s    r   Úis_symboliczNestedIntNode.is_symbolici   r)   r   c                 ó   — | j         S r
   )r   r   s    r   Ú
nested_intzNestedIntNode.nested_intl   s
   € ØŒyÐr   c                 ó   — dS r'   r    r   s    r   r   zNestedIntNode.is_constanto   r)   r   Únumc                 ó”   — t          |¦  «        t          ur$t          dt          |¦  «        j        › �¦  «        ‚t	          |¦  «        S )Nz Expected num to be int, but got )r@   ÚintÚAssertionErrorÚ__name__r   )r   r[   s     r   Úwrap_intzNestedIntNode.wrap_intr   sI   € Ý�‰9Œ9�CÐÐÝ ØGµ4¸±9´9Ô3EÐGÐGñô ð õ ˜sÑ#Ô#Ð#r   )r   r   ) r_   Ú
__module__Ú__qualname__r]   r   r   ÚOptionalr!   Úboolr%   r(   r+   r-   r/   ÚAnyr3   r6   r9   r;   r=   rB   rH   rK   rO   rQ   rS   rU   rW   rY   r   r   r`   r    r   r   r   r   %   s  € € € € € ð˜Sð ¨ð °ð ð ð ð ð #ð ð ð ð ð˜h sœmð ð ð ð ð˜ð ð ð ð ð˜$ð ð ð ð ð˜ð ð ð ð ð˜tð ð ð ð ðð ð ð ð,�cð ,ð ,ð ,ð ,ð
�Sð ð ð ð ð˜ð ð ð ð ð˜#ð ð ð ð ð˜Sð ð ð ð ð<˜ð < ð <ð <ð <ð <ðE˜ð E ð Eð Eð Eð EðI˜ð I ð Ið Ið Ið IðI˜ð I ð Ið Ið Ið IðI˜ð I ð Ið Ið Ið IðE˜ð E ð Eð Eð Eð EðE˜ð E ð Eð Eð Eð Eð˜Tð ð ð ð ð˜Cð ð ð ð ð˜Tð ð ð ð ð$˜Cð $ Oð $ð $ð $ð $ð $ð $r   )
ÚtypingrE   Ú'torch.fx.experimental._constant_symnoder   Ú__all__re   rd   r   r   r   r    r   r   ú<module>ri      sÄ   ðØ Ð Ð Ð à €€€Ø CÐ CÐ CÐ CÐ CÐ Cð Ð
€ðˆSð �sð ˜tð ð ð ð ð/ˆSð /�sð /˜tð /ð /ð /ð /ð"R$ð R$ð R$ð R$ð R$ñ R$ô R$ð R$ð R$ð R$r   