VhdZddlZddlZddlZddlmZmZddlmZddl m Z m Z ddl m Z mZddlmZmZdd lmZmZdd lmZdd lmZmZmZmZd d lmZmZd dlm Z erddl!m"Z"dZ#GddeZ$GddeZ%Gdde$Z&Gdde$Z'Gdde$Z(Gdde(Z)Gdde(Z*Gdd eZ+Gd!d"e+Z,Gd#d$e+Z-y)%a Dictionary-related variable tracking classes for PyTorch Dynamo. This module implements variable tracking for different types of dictionary-like objects: - Regular Python dictionaries (dict) - Ordered dictionaries (collections.OrderedDict) - Default dictionaries (collections.defaultdict) - Dictionary views (keys and values) - Sets and frozensets (implemented internally using dictionaries) These classes are responsible for tracking dictionary operations during graph compilation, maintaining proper guards for dictionary mutations and key existence checks. They handle dictionary creation, modification, key/value access, and view operations while ensuring correct behavior in the compiled code through appropriate guard installation. The implementation uses a special _HashableTracker wrapper to handle dictionary keys while preserving proper aliasing semantics. Sets are implemented as dictionaries with None values for efficiency and code reuse. N)Optional TYPE_CHECKING)is_fake) polyfills variables)create_call_functioncreate_instruction)raise_observed_exception unimplemented) GuardBuilder install_guard)is_from_local_source)cmp_name_to_op_mapping dict_keys dict_valuesspecialize_symnode)ValueMutationNewVariableTrackerConstantVariable)InstructionTranslatorct|tjr!|js|j ryt|tj r5|j jjjdduSt|tjrtd|jDSt|tjtjtjtj tj"j$tj&tj(tj*j,tj.tj0tj2tj4tj6tj8tj:fS)NT example_valuec32K|]}t|ywN) is_hashable).0es M/home/dcms/DCMS/lib/python3.12/site-packages/torch/_dynamo/variables/dicts.py zis_hashable..Cs3a;q>3s) isinstancerLazyVariableTracker is_realizedrTensorVariableas_proxynodemetaget TupleVariableallitemsBuiltinVariableSymNodeVariabler EnumVariable user_definedUserDefinedClassVariableUserFunctionVariableSkipFunctionVariablemisc NumpyVariableNNModuleVariableUnspecializedNNModuleVariableMethodWrapperVariableTorchInGraphFunctionVariableTypingVariableFunctoolsPartialVariableWeakRefVariable)xs r!rr1s. 1i334  MMO!Y--.zz|  %%))/:$FF Ay.. /3177333 ))))**&&&&??....,,**77//66((22))   ceZdZdhejZGddZefdeeefddffd ZdZd Z d Z d Z d Z de fd ZdZdZdZdZdddefdZdddefdZdefdZdefdZdZdZ dfd ZdZdZfdZxZS) ConstDictVariableuser_clscNeZdZdZd dZedZdZedZ ddde fd Z y) "ConstDictVariable._HashableTrackeraB Auxiliary opaque internal class that wraps a VariableTracker and makes it hashable This should not be seen or touched by anything outside of ConstDictVariable and its children Note that it's also fine to put VTs into dictionaries and sets, but doing so does not take into account aliasing returnNcrt|}t|stdt|d|||_y)NzDict key of type z. Key: )rrr typevt)selfrHs r!__init__z+ConstDictVariable._HashableTracker.__init__hs6#B'Br? 1$r(72$GHDGr?ct|jtjrN|jj s4|jj r|jj St|jtjr3|jjjjd}|St|jtjr:tjtfd|jjD}|St|jtj r|jj"St|jtj$r|jj"St|jtj&r|jj)St|jtj*r6tj|jj,j.S|jj1}|S)Nrc3BK|]}|jywr)underlying_value)rr Hashables r!r"zFConstDictVariable._HashableTracker.underlying_value..}sN1(1+66Ns)r#rHrr$r%roriginal_valuer&r'r(r)r+rA_HashableTrackertupler-r7valuer8r3 get_functionr= referent_vtrMas_python_constant)rIr>rNs @r!rMz3ConstDictVariable._HashableTracker.underlying_valueqs477I$A$AB++-GG'')ww--//$''9#;#;<GG$$&++00A HDGGY%<%<=,==N NNHDGGY%?%?@ww}}$DGGY%L%LMww}}$DGGY%C%CDww++--DGGY%>%>?,== 3 34EEEGG..0Hr?c,t|jSr)hashrMrIs r!__hash__z+ConstDictVariable._HashableTracker.__hash__s--. .r?c t|t|k7ryt|trGtjt |t |k(xrt fdt||DSt|r||uS||k(S)NFc3HK|]\}}j||ywr)_eq_impl)ruvrNs r!r"z>ConstDictVariable._HashableTracker._eq_impl..s'0041H%%a+0s") rGr#rQrArPlenr,zipr)abrNs @r!r\z+ConstDictVariable._HashableTracker._eq_impls|Aw$q'!Au%,==1vQ'C08;Aq 0-Av Av r?otherc tj}t||s%tj|sJt |t||r&|j |j|jS|j |j|Sr)rArPr#r is_literalrGr\rM)rIrcrNs r!__eq__z)ConstDictVariable._HashableTracker.__eq__s(99HeX.2B2M2Me2T U  T%*(()>)>@V@VWW$$T%:%:EB Br?rEN) __name__ __module__ __qualname____doc__rJpropertyrMrY staticmethodr\boolrfr?r!rPrDasP      4 /     C D C Cr?rPr-rENc d|vr|jdd|vr|jdt|di|tjt fd|j DsJfd}|j Dcic]\}}|||c}}|_t|j |_ |j|_ ||_ ycc}}w)Noriginal_itemsshould_reconstruct_allc3lK|]+\}}t|tfxrt|t-ywr)r#r)rr>r^rNs r!r"z-ConstDictVariable.__init__..s? 1 q?H5 6 /1o. / s14c0t|r|S|Sr)r#)keyrNs r! make_hashablez1ConstDictVariable.__init__..make_hashables$S(33 F# Fr?ro) popsuperrJrArPr,r-rsourcerrcopyrqrB) rIr-rBkwargsrvr>r^rN __class__s @r!rJzConstDictVariable.__init__s v % JJ' ( #v - JJ/ 0 "6"$55       G7**:J:J:LM$!Q -MMM3Acnddjd|jjDzdzS)N{, c3|K|]4\}}|jjd|j6yw)z: NrH debug_repr)rrr^s r!r"z/ConstDictVariable.debug_repr..s7=AQ144??$%R '78s:<})joinr-rXs r!rzConstDictVariable.debug_reprs? iiEIZZEUEUEW    r?c|jjDcic].\}}|jj|j0c}}Scc}}wr)r-rHrUr~s r!rUz$ConstDictVariable.as_python_constantsP ((* 1 DD # # %q';';'= =   rc|j|jjDcic] \}}|jj|"c}}Scc}}wr)install_dict_keys_match_guardr-rHrUr~s r!keys_as_python_constantz)ConstDictVariable.keys_as_python_constantsE **,9=9I9I9KLA'')1,LLLs%Ac|jSr)rBrXs r! python_typezConstDictVariable.python_types }}r?ct|tsJtj}t |xrD|||j vxr.t|j ||t j Sr)r#rrArPrr-rDeletedVariable)rIrHrNs r! __contains__zConstDictVariable.__contains__se"o...$55 O T  * Ttzz(2,79R9RSS r?c t|jjDcgc]}t|tj s| c}Scc}wr)r_r-valuesr#rrrIr>s r!r_zConstDictVariable.lensI**, !!Y%>%>?    s#A cvjrytfdjjDS)NTc3K|]<\}}jjj|j|>ywr) is_new_itemrqr*rH)rrurRrIs r!r"z2ConstDictVariable.has_new_items..s? U   T0044SVV->-@emmo&"U]]_*== =%yBuI%%r?c8|jtjurjfdd}|jj D]h\}}|j j |j}|j||s |jsJ|j||dz }j|jtjur*jtd|gtddyjtd|y)NcnjjtjdgS)N OrderedDict) extend_outputcreate_load_python_module collectionscreate_load_attrcodegensr!z/ConstDictVariable.reconstruct..s3--99+F00?r?rr BUILD_MAPargF)rBrr add_push_nullr-rqr*rHrrrrr r append_output)rIrnum_argsrurRitems ` r! reconstructzConstDictVariable.reconstructs ==K33 3  ! ! ****, JC&&**3662De,0K0KA    ==K33 3  ! !&{A)!U3   ! !"4[h"O Pr?txrrctj|}||jvrtt||j|Sr)rArPr-r KeyErrorrIrrrus r!'getitem_const_raise_exception_if_absentz9ConstDictVariable.getitem_const_raise_exception_if_absent%s8 005 djj $Xr 2zz#r?ctj|}||jvrtd|j|j|S)Nzdict KeyError: )rArPr-r rRrs r! getitem_constzConstDictVariable.getitem_const-s?005 djj OCII;7 8zz#r?chtj|}||jvry|j|Sr)rArPr-)rIrrus r!maybe_getitem_constz%ConstDictVariable.maybe_getitem_const3s/005 djj zz#r?c@||vsJtj|}t|jj j |}t|j j |}t|tjr|jyyr) rArPrQr-keysindexrqr#rr$r)rIrruroriginal_key_vts r!realize_key_vtz ConstDictVariable.realize_key_vt9s~d{{005djjoo'(..s3 3 3 8 8 :;EB oy'D'D E  # # % Fr?cn|jr)t|jtjyyr)ryr make_guardr DICT_KEYS_MATCHrXs r!rz/ConstDictVariable.install_dict_keys_match_guardBs% ;; $//,*F*FG H r?c ddlm}|jsy|jjj |ry|d|v}|dj\t |d|rMt|jtjtj|dj| y|djr(|r|j|dy|jyy)Nrrr)ruinvert)rryoutput side_effects is_modifiedr#rr functoolspartialr DICT_CONTAINSrRrr)rIrargsrcontainss r!install_dict_contains_guardz-ConstDictVariable.install_dict_contains_guardFs '{{  99 ! ! - -d 3 7d? 7>> !ja:J&K %%$22 GMM#+| !W^^##DG,224 r?c ddlm}m}m}tj }|xrt |d} |dk(rtjtj|g|i|} |jjj||jj| j|jdS|dk(r%t!|dk(sJ|j#||dS|dk(r|s|rJ|j%|j&r=|jj(j+|j&j-||jjD cgc]\} } || j.| gc} } S|dk(rj|j%|j&r=|jj(j+|j&j-|s|rJt1|S|dk(rj|j%|j&r=|jj(j+|j&j-|s|rJt3|S|d k(rJ|j%|s|rJ|j5|jj7t9d S|d k(r;|s|rJ|j%|jt!|jS|d k(r| r|j;rt|j%|st!|d k(sJ|jjj||d|j||d<|jdS|dk(r| r|j;rr|j%d|_|jjj||jj?||d|jdS|dvrVt!|dvrI|d|vrB|jA||t!|dk(r|dk(rtCtD||dS|dS|dk(rb| r`|j;rPd|_|jjj||jjG||dS|dk(rXd|_|jjj||jjI|jdS|dk(rM|j;r<|j%t!|dk(} t!|dkD}| s|r|jjj|| rktK|dtr|dj%|d}n|jM|t|d}|jj|j|rT|jD cic]\} } ||j| |  }} } |jj||jdStN|||||S|dvr|d|vr|jS||dS|dk(r9t!|dk(r+|jA|||d|v}|j|S|dk(r| r|j;r|j%|rJt!|d ksJ|jU|d}||St!|dk(r|jd}n|d}|jjj|||j||d<|S|dk(r|j%|st!|dk(sJ|jjj|||d}|j|}|jjG|||j|<|jdStN|||||Scc} } wcc} } w)Nr)r.rr+rrJ __getitem__r-rrrz)r- mutation_typery__len__ __setitem__r __delitem__T)rwr*)rrrwclearupdate)r* __getattr__r setdefault move_to_end)+rr.rr+rArPrrdict call_dictrrmutationr-rcreater_rrryguard_on_key_orderaddnamerHDictKeysVariableDictValuesVariableclonerzr is_mutablerrrrr rrwrr#call_custom_dictrx call_methodrr)rIrrrr{r.rr+rN arg_hashable temp_dict_vtrr^has_arg has_kwargsdict_vtrrRr>ruvalr|s r!rzConstDictVariable.call_methodmsn GF$554 DG 4 : D944T:DD#L II " " + +D 1 JJ  l00 1*#**40 0 ] "t9> !>??DGL L W_ ''  . . 0{{ ,,001A1A1CD 6:jj6F6F6HIdaay)I V^  . . 0{{ ,,001A1A1CD ''#D) ) X   . . 0{{ ,,001A1A1CD ''%d+ + V^  . . 0 ''::jjoo'7G7IRV Y  ''  . . 0*#**3tzz?; ; ] "|8I  . . 0#d)q. 00 II " " + +D 1,0GDJJxQ( )*#**40 0 ] "|8I  . . 0*.D ' II " " + +D 1 JJ " "8DG#4 5*#**40 0 ^ #D V(;Qt@S  , ,R 64yA~5=,Xr:'--Aw U]|0A*.D ' II " " + +D 1::>>(47"34 4 W_*.D ' II " " + +D 1 JJ   *#**40 0 X $//"3  . . 0$i1nGVqJ* &&//5!$q'+<=Q==?"&q'"1"B"B2tTRSW"UJJ%%gmm4%+LLN Aq!!8!1!8!8!; !>,,T!W5E  t9>/(//5AQA &&//501 8DG,- ] "  . . 0#d)q. 00 II " " + +D 147#C**S/C JJNN3 !DJJsO*#**40 07&r4v> >cJRs )a #ac|j|jjDcgc]}|jc}Scc}wr)rr-rrH)rIrr>s r!unpack_var_sequencez%ConstDictVariable.unpack_var_sequences1 **,"jjoo/0000sAc|jturB||jjvrtjdStjdSt d|jdy)NTFz hasattr on z is not supported)rBr__dict__rrr )rIrrs r!call_obj_hasattrz"ConstDictVariable.call_obj_hasattr sZ ==D t}}---'..t44#**51 1 DMM?2CDEr?c B|jt|di|SNro)rrxr)rIr{r|s r!rzConstDictVariable.clones! **,w}&v&&r?rzlist[VariableTracker]r{zdict[str, VariableTracker]rEr)rhrirjr_nonvar_fieldsrPrrJr'rrUrrrnrr_rrrrrrrrrrrrr __classcell__r|s@r!rArA[s  ' 'N FCFCV !O_45 !  !DN  M $   & QB)0? 7o  &/&I%5NV?& V? - V?  V?p1F''r?rAcXeZdZdeddffd ZdZdZdedd ee dfddfd Z xZ S) MappingProxyVariabledv_dictrENc Vt|di|t|tsJ||_yr)rxrJr#rArrIrr{r|s r!rJzMappingProxyVariable.__init__s+ "6"'#4555 r?c8|jj|Sr)rr)rIrs r!rz(MappingProxyVariable.unpack_var_sequences||//33r?c|jr tdjfd|jj t ddy)NzgCan't reconstruct an existing mapping variable because the connection to the original dict will be lostcnjjtjdgS)NMappingProxyType)rrtypesrrsr!rz2MappingProxyVariable.reconstruct..)s4G))55e<,,-?@r?rF)ryr rrrr rIrs `r!rz MappingProxyVariable.reconstruct!sQ ;; D      21e<=r?rrr{c|jr/|jjjr t d|j j ||||S)NaA dict has been modified while we have an existing mappingproxy object. A mapping proxy object, as the name suggest, proxies a mapping object (usually a dict). If the original dict object mutates, it is reflected in the proxy object as well. For an existing proxy object, we do not know the original dict it points to. Therefore, for correctness we graph break when there is dict mutation and we are trying to access a proxy object.)ryrrhas_existing_dict_mutationr rr)rIrrrr{s r!rz MappingProxyVariable.call_method3sJ ;;29911LLN 7 ||''D$??r?) rhrirjrArJrrlistrstrrrrs@r!rrsY 1 4>$@$% @ S++, @  @r?rceZdZdZdZy)NNModuleHooksDictVariablecyrrorXs r!rz7NNModuleHooksDictVariable.install_dict_keys_match_guardI r?cyrrorIrrs r!rz5NNModuleHooksDictVariable.install_dict_contains_guardLrr?N)rhrirjrrror?r!rrGs   r?rc^eZdZddfd ZfdZfdZedZ dfd ZxZ S) DefaultDictVariablec ^t|||fi||tjusJ||_yr)rxrJr defaultdictdefault_factory)rIr-rBrr{r|s r!rJzDefaultDictVariable.__init__Qs3 3F3;22222.r?ct|jtttfvr |jsyt |S)NF)rrrQrr-rxis_python_constantrIr|s r!rz&DefaultDictVariable.is_python_constantVs1   eT': :4::w)++r?c^d|jjdt|dS)Nz defaultdict(r))rrrxrs r!rzDefaultDictVariable.debug_repr]s44//::<=R@R@T?UUV W r?ct|tjr"|jtt t tfvSt|tjjSr) r#rr.fnrrQrset functionsBaseUserFunctionVariablers r!is_supported_argz$DefaultDictVariable.is_supported_argbsC c944 566dE455 5c9#6#6#O#OP Pr?c,|dk(r}t|dk(sJ|d|vr|j||dS|jt|d|jj |gi}t ||d|d|f||St |||||S)Nrrrr)r_rrr call_functionrxr)rIrrrr{ default_varr|s r!rzDefaultDictVariable.call_methodis = t9> !>Aw$))"d1g66''/"d1gY00"&"6"6"D"DRR"PKG'MDG[+A6'&7&r4v> >r?rrgr) rhrirjrJrrrmrrrrs@r!r r PsP/ , QQ ?& ? - ?  ??r?r ceZdZdZdeeddffd ZdZedZ e dZ d Z d Z d Zd Zd eedeeefddffd ZdddefdZdZdZxZS) SetVariablez-We model a sets as dictonary with None valuesr-rENc vtj|tj}t ||fi|yr)rfromkeysr_default_valuerxrJrIr-r{r|s r!rJzSetVariable.__init__s0  e[%?%?%AB )&)r?c|jsyddjd|jjDzdzS)Nzset()r,c3PK|]}|jj ywrrrrs r!r"z)SetVariable.debug_repr..!O!$$//"3!O$&rr-rrrXs r!rzSetVariable.debug_reprs7zz!OTZZ__=N!OOORUU Ur?cHt|jjSr)rr-rrXs r! set_itemszSetVariable.set_itemss4::??$%%r?c,tjdSr)rrror?r!r!zSetVariable._default_values &&t,,r?cp|jDchc]}|jjc}Scc}wr)r+rHr'rIrs r!r'zSetVariable.as_proxys$)-8A 888!3ctSr)rrXs r!rzSetVariable.python_types r?cp|jDchc]}|jjc}Scc}wrr+rHrUr.s r!rUzSetVariable.as_python_constant'37>>Ba'')BBBr/c|j|jDcgc]}|jc}|jt dt |jycc}w)N BUILD_SETr)foreachr+rHrr r_rIrr>s r!rzSetVariable.reconstructsEt~~6!670#dnnBUVW7sA$rr{rc|dk(r1|rJt|dk(sJd}|dtjf}n|dk(rA|rJ|rJ|jj j }t ||||f||S|dk(rM|rJt|dk(sJtjtjj|||dgiS|dk(rM|rJt|dk(sJtjtjj|||dgiS|dk(rM|rJt|dk(sJtjtjj|||dgiS|d k(rM|rJt|dk(sJtjtjj|||dgiS|d k(rkt|dk(r]|j!rM|rJt|dk(sJtjtj"j|||dgiS|d k(r8|rJt|dk(sJ|d|vr t%d t ||d||S|d k(rC|rJt|dk(sJ|d|vrt ||d||St'j(dSt |||||S)Nrrrrrw isdisjoint intersectionunion differencerremovezkey does not existdiscard)rR)r_rr!r+rwrHrxrrr3rset_isdisjointrset_intersection set_unionset_differencer set_updater rr)rIrrrr{resultr|s r!rzSetVariable.call_methods 5= :t9> !> DG[779:D U] :O8^^''),,F G D6)V <M \ ! :t9> !>11((mBtAw4 5^ # :t9> !>11**mBtAw4 5W_ :t9> !>11)2E2EFTTT47OR \ ! :t9> !>11((mBtAw4 5X #d)q.T__5F :t9> !>11)2F2FGUUT47OR X  :t9> !>Awd"237&r5$? ? Y  :t9> !>Aw$w*2udFCC'..T::w"2tT6::r?rrrctd)NzIllegal to getitem on a set) RuntimeError)rIrrs r!rzSetVariable.getitem_consts899r?cyrrorXs r!rz)SetVariable.install_dict_keys_match_guard r?cyrror s r!rz'SetVariable.install_dict_contains_guardrHr?)rhrirjrkrrrJrrlr+rmr!r'rrUrrrrrrrrrs@r!rrs7*O$*  *V &&--9CX?;?# ?; S/)* ?;  ?;B: 7:o:  r?rc~eZdZdeeddffd ZdZedZdZ dZ d Z d eed e e efdd ffd ZxZS)FrozensetVariabler-rENc &t||fi|yrrxrJr"s r!rJzFrozensetVariable.__init__ )&)r?c|jsyddjd|jjDzdzS)Nz frozenset()rr$c3PK|]}|jj ywrrr&s r!r"z/FrozensetVariable.debug_repr..r'r(rr)rXs r!rzFrozensetVariable.debug_reprs7zz !OTZZ__=N!OOORUU Ur?c6|jjSr)r-rrXs r!r+zFrozensetVariable.set_itemsszz  r?ctSr) frozensetrXs r!rzFrozensetVariable.python_type r?cp|jDchc]}|jjc}Scc}wrr2r.s r!rUz$FrozensetVariable.as_python_constantr3r/cj|jDcgc]}|jc}jfdj t ddycc}w)NcFjjdgS)NrS)rcreate_load_globalrsr!rz/FrozensetVariable.reconstruct..s%G))..{;r?rF)r6r+rHrrr r7s ` r!rzFrozensetVariable.reconstructsRt~~6!67   21e<=7sA%rr{rcN|dvrtd|dt| ||||S)Nrrwrr=r>rIllegal call_method z on a frozensetrFrxrrIrrrr{r|s r!rzFrozensetVariable.call_method9 I I!5dV?KL Lw"2tT6::r?)rhrirjrrrJrrlr+rrUrrrrrrs@r!rKrKs*O$*  *V !!C > ;?# ; S/)* ;  ; ;r?rKcxeZdZdeeddffd ZdZedZdZ dZ d eed e e efdd ffd Z xZS) DictKeySetVariabler-rENc &t||fi|yrrMr"s r!rJzDictKeySetVariable.__init__)rNr?c|jsyddjd|jjDzdzS)Nz dict_keys([])z dict_keys([r$c3PK|]}|jj ywrrr&s r!r"z0DictKeySetVariable.debug_repr..6sH144??,Hr(z])r)rXs r!rzDictKeySetVariable.debug_repr0sAzz"((Hdjjoo6GHHI r?c|jSr)r-rXs r!r+zDictKeySetVariable.set_items:s zzr?ctSrrrXs r!rzDictKeySetVariable.python_type>rTr?ctj|jDchc]}|jj c}dj Scc}wr)rr r+rHrUrr.s r!rUz%DictKeySetVariable.as_python_constantAs>}}04 ?1QTT $ $ & ? $&  ?s!Arr{rcN|dvrtd|dt| ||||S)NrZr[z on a dict_keysr\r]s r!rzDictKeySetVariable.call_methodFr^r?)rhrirjrrrJrrlr+rrUrrrrrs@r!r`r`(sy*O$*  * ;?# ; S/)* ;  ; ;r?r`ceZdZUdZdZeeed<deddffd Z e dZ e dZ d Z d Zd ed d eed fdd ffd ZxZS)DictViewVariablezu Models _PyDictViewObject This is an "abstract" class. Subclasses will override kv and the items method NkvrrEc vt|di||jdvsJt|tsJ||_y)N)rrro)rxrJrkr#rArrs r!rJzDictViewVariable.__init__[s> "6"ww,,,,'#4555 r?c`t|jj|jSr)getattrrr-rkrXs r! view_itemszDictViewVariable.view_itemsas"3wt||))477355r?ctr)NotImplementedErrorrXs r! view_items_vtzDictViewVariable.view_items_vtes "!r?cXfd}jDcgc] }|| c}Scc}w)Nc>jdk(r |jS|S)Nr)rkrH)r>rIs r!unwrapz4DictViewVariable.unpack_var_sequence..unwrapls77f,144 3! 3r?)ro)rIrrur>s` r!rz$DictViewVariable.unpack_var_sequenceks& 4$(??3aq 333s'c||j|j|j|jdy)Nr)r load_methodrkrrs r!rzDictViewVariable.reconstructqs. DGG$Ar?rrr{cn|dk(r|jj||||St|||||S)Nr)rrrxr]s r!rzDictViewVariable.call_methodvs> 9 <<++BdFC Cw"2tT6::r?)rhrirjrkrkrr__annotations__rArJrlrorrrrrrrrrs@r!rjrjRs B  1 66"" 4  ;$% ; S++, ;  ; ;r?rjcfeZdZdZedZedZdZdedde e dfddffd Z xZ S) rrc,t|jSr)rrorXs r!r+zDictKeysVariable.set_itemss4??##r?cT|jDcgc]}|jc}Scc}wr)rorHrs r!rrzDictKeysVariable.view_items_vts #oo....s%ctSrrfrXs r!rzDictKeysVariable.python_typerTr?rrr{rEcT|dk(r|jj||||S|tvrkt|dtt fst jtSt jt||j|djSt|||||S)Nrr) rrrr#rrrrNotImplementedr+rxr]s r!rzDictKeysVariable.call_methods > !<<++BdFC C ) )d1g 5E'FG'..~>>#**&t,T^^T!W=N=NO w"2tT6::r?) rhrirjrkrlr+rrrrrrrrrs@r!rrsl B $$//;$% ; S++, ;  ;;r?rc&eZdZdZedZdZy)rrc,t|jSr)rrorXs r!rrz DictValuesVariable.view_items_vtsDOO$$r?ctSr)rrXs r!rzDictValuesVariable.python_typesr?N)rhrirjrkrlrrrror?r!rrs B %%r?r).rkrrrtypingrrtorch._subclasses.fake_tensorrrrrbytecode_transformationr r excr r guardsr rryrutilsrrrrbaserrconstantrtorch._dynamo.symbolic_convertrrrArrr rrKr`rjrrror?r!rs( *1#N90)VV3&D' Ty'y'x -@?-@` 1 /?+/?jq #q h,; ,;^';';T-;-;`;';B ) r?