sh F SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKJr SSK J r SSKJ r SSK J r JrJrJrJrJr SSKJr SSKJr SSKJrJrJr SS KJr SS KJr SS KJ r J!r! SS K"J#r#J$r$ SS K%J&r& \(a SSK'r'SSK(J)r) SSK*r*SSK+J,s J-r. SSK/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;Jr> SSK?J@r@ SSKAJBrB SSKJCrC SSKDJErEJFrF SSKGJHrH SSKIJJrJ SSKKJLrL SSKMJNrN SSKOJPrPJQrQJRrRJSrSJTrTJUrUJVrVJWrWJXrXJYrYJZrZJ[r[J\r\ /SQr]\\*RR/\\L4r`\R"SS55rb\R"SS55rcSrdS \\eS!\.RS"\\eS#\.RS$\g4 S%jrh\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR/ rw\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRR\*RRGR/ r\ S@S&j5r\ S'5rS(\\*GR GR \4S$\\\*GR GR \4S)44S*jrSAS+jrS,\\*GR GR \4S-\\*GR GR \4S.\\4S/jrS0\*RRS1\SS$\\*RR\S44S2jrS0\*RR4S3jrS0\*RRS4\\c4S5jrS,\\*GR GR \4S-\\*GR GR \4S.\\S6\g4S7jr"S8S95rS:rSBS0\*RRS;SjjrS?rg)CN) namedtuple)Iterator)contextmanager)AnyCallablefinalOptional TYPE_CHECKINGUnion)autograd_not_implemented)FakeScriptObject)_deregister_op_impl_is_op_registered_to_fake_ruleregister_op_impl)FakeTensorMode)#first_call_function_nn_module_stack)_PyTreeCodeGen _PyTreeInfo)immutable_dictimmutable_list)insert_deferred_runtime_asserts) ValueRanges)_collect_all_valid_cia_ops_collect_and_set_constant_attrs_collect_param_buffer_metadata_detect_fake_mode_from_gm_fakify_params_buffers_get_decomp_for_cia_is_preservable_cia_op_name_hoo_subgraph_placeholders7_override_graph_signature_for_temp_registered_constants/_overwrite_signature_for_non_persistent_buffers)_populate_param_buffer_metadata_to_new_gm_register_constants_as_buffers_rename_without_collisions_special_op_to_preserve_ciaplaceholder_naming_pass)Verifierdetect_fake_modeunset_fake_temporarily) is_equivalentreorder_kwargsCustomDecompTable) compatibility) PassResult) PassManager) ArgumentSpecConstantArgumentCustomObjArgumentExportGraphSignature InputKind InputSpec OutputKind OutputSpecSymBoolArgumentSymFloatArgumentSymIntArgumentTensorArgument TokenArgument)ExportedProgramModuleCallEntryModuleCallSignaturedefault_decompositionsc\rSrSr%\\\S'\\\S'\R\S'\R\S'Sr \ \\ \S'Sr S r g) rD^inputsoutputsin_specout_specNforward_arg_namescURH0nURUR:XdMURUlM2 URH0nURUR:XdMURUlM2 gN)rHnamerI)self original_nodenew_nodeios q/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/export/exported_program.pyreplace_all_uses_with)ModuleCallSignature.replace_all_uses_withfsZAvv+++!Avv+++!)__name__ __module__ __qualname____firstlineno__listr5__annotations__pytreeTreeSpecrLr strrV__static_attributes__rYrXrUrDrD^sE   ,  __oo-1xS *1'rXrDc4\rSrSr%\\S'Sr\\\S'Sr g)rCofqnN signaturerY) rZr[r\r]rbr_rgr rDrcrYrXrUrCrCos H/3Ix+,3rXrCcF^[R"T5U4Sj5nU$)Nc^>[5 T"U0UD6sSSS5 $!,(df  g=frNr+)argskwargsfns rUwrapper/_disable_prexisiting_fake_mode..wrappervs" # %t&v&& % %s ,) functoolswraps)rlrms` rU_disable_prexisiting_fake_moderqus%__R'' NrX spec1_type spec1_context spec2_type spec2_contextreturnc.UbUcXL=(a X:H$[U[[45(a[U[[45(aX:H$[U[[45(a[U[[45(aX:H$XL=(a X:H$)z]Treat containers and their immutable variants as the same type. Otherwise compare as normal. ) issubclassdictrr^r)rrrsrtrus rU_fx_collection_equivalence_fnrz~sZ/'JM,JJ*t^455*T>*;;--*t^455*T>*;;--  # F (FFrXc ## 0n[5nUR5GHupEURR5X$'UR U5 [ H3nXdR;dMUR U5"[USS95 M5 [U5n[RRRUR;a/UR[RRR U(a9UR [RRR5"U5 Sn[U5(d$[U5"[R "UUS95 ["H+n XR;dMUR U 5"U5 M- GM Sv UH_n U RR%5 U RR'X*5 U R(R%5 [+U 5 Ma g!UH_n U RR%5 U RR'X*5 U R(R%5 [+U 5 Ma f=f7f)NT)deferred_errorc\USnUS U U"U0UD6sSSS5 $!,(df  g=f)Noriginal_callablerY)fake_tensor_modeoprjrkorig_cia_callables rU$_force_dispatch_to_orig_cia_callableQ_override_composite_implicit_decomp.._force_dispatch_to_orig_cia_callables3 &': ; *+!($9&9"!!s +)r~)setitems py_kernelscopyadd(_AUTOGRAD_ALIAS_BACKEND_KEYS_TO_OVERRIDEpy_implr rtorch_C DispatchKeyCompositeImplicitAutogradrrropartial_BACKEND_KEYS_TO_OVERRIDEclearupdate_dispatch_cacher) cia_ops_to_callablesafe saved_tables patched_ops op_overloaddecomp_callableoverride_dispatch_keyrrkeyrs rU#_override_composite_implicit_decomprs L%K(;(A(A(C$ $/$:$:$?$?$A ! $%M !$,B,BB##$9:,[N&N0 < 88   9 9[=S=S S&&uxx';';'U'UV     4 4 N N O  : .k:: [ )!!8&7 -C000 ##C():;#-[)D@$ B MM   ! MM !1 2    $ $ &  # +B MM   ! MM !1 2    $ $ &  # s-A&J,DJ J-H1A&JA'I>>Jc#$# [[RRRR [ [RRRR[ 0SS9 Sv SSS5 g!,(df  g=f7f)NF)r)rropsatento dtype_layoutr&dtyperYrXrU!_override_decomp_aten_to_variantsrsa - IINN   * *,G IINN   # #%@       sA/B1A?6 B? B  B decomp_table.c^[[55n0n[UR55HhnX1;aXX#'UR U5 X M#[ U5(dM5UR 5nURS5(aS5eXX#'Mj UH n[X%'M X 4$)NrzThis should be a custom op) rrr^keysremoverrO startswithr&)rall_preservable_cia_opsrrop_nameks rU,_split_decomp_table_to_cia_and_python_decompr's""<">?<$$&'$  (&2&6  # # * *2 . $B ' 'ggiG))&11 O3O O1&2&6  #9(@%!<%  ,,rXc[5$)z This is the default decomposition table which contains decomposition of all ATEN operators to core aten opset. Use this API together with :func:`run_decompositions()` r/rYrXrUrErESs  rX cia_to_decomppython_decomp_tablejoint_loss_indexcSSKJn SSKJn SSKJnJnJn Jn J n J n SSK J n SnU"X5(GdUR5n0[URSS95E[UR!SS95EnSS KJn U"U5 0[URSS95E[UR!SS95En['UR(5nUc[+U "5S S 9nUR,nUR.R0R2R4nUR6[8[:4;a[<R>"[:SU/5n[A[CUURDU55UR.lURG5 [IURJURLU5 [OXRPURJRR5n[UUU5n[WU5nSS K,J-nJ.n /nUR.R^HnUR`S :XdM[cURdS[f5(aySnURdSRhc'URLURdSRjnO#URdSRhRlnURoU5 MURoURdS5 M U [q5 [sU5 U"URt5 [<Rv"UURD5nU"U/USQUSRy5Q70U5unn n!n"n#U"UU U!UU"USSUS9 n$U$RLR{5V%V&s0sH#un%n&U%[cU&[|5(aU#U&OU&_M% sn&n%U$l&U$R~n'U$Rn([U(U5n([UU'U(5 [URJU(5n(U"U'U(U"5n)[U'U(UU U!UU)5 SSS5 SSS5 SSS5 SSS5 SSS5 U "W'5 U "U'5 U "U'W(5 [U'U(5un'n(UR{5Hun*n+U*U;dMU+URPU*'M UR{5Hvun*n+U*URP;a+[cU+[RR5(aeU*URP;dMSU*U;dM[URPRU*5 Mx U'U(URP4$UR(R.R^Vs/sHnUR`S :XdMUPM n,nU,Vs/sHnURdSPM n-nUR(R!5V*V.s/sHun*n.U*PM n/n*n.U/Hn*[UR(U*5 M [U-5nUc[R"5OUnU(a[ROUn0U "5 U [sU5 U0"5 U"UR(U-UUbS OSUbUOSS9un'n1U'R.R5 SSS5 SSS5 SSS5 SSS5 Sn2W'R.R^Vs/sHnUR`S :XdMUPM n3n[9U'R.R^5SRSn4[U35[U,5:Xde[U,U35Hun5n6U5Rj=U6l5U6lRM U3V7s0sHn7U7RjU7Rj_M n8n7U'R.R^H;nUR`S :XaM[U8URjURj5Ul5M= [U'5 SSKUJVn9JWn: SSKXJYn; [RRR(dWSn<[U'5n=U=bGU:"U'[R"U9Uun>n*[cU4U>[RR5(dM4U*U4U>l5M@ [URJR5V>V?s0sHKun>n?[cU?R[5(aM'U?RRjU3U>Rj_MM n@n>n?[URJR5V>V?s/sHFun>n?[U?RU2"U?RU3U>5U?RU?R5PMH nAn>n?/nB[W1R5V>V*s0sH un>n*U*U>_M nCn>n*[9U1RR55nDUDU4S[UD5Vs/sHnURjPM sn:XdeU1RR{5HunEnFWCUFn>URJRU>nGUGR[R[R4;deWGR[R:Xa[RO[RnHWGR[R:Xa WGROWGRRjnIWBRo[WH[WES9UIS95 M [URJR5Hun>n?WBRo[U>U:Xa[RO U?RU2"U?RU4[WD5U>-5W@RU?RU?R555 M UGbU1RceU1RRnJ[U1R5[URJR5:Xde[URJR5V>V?s0sH7un>n?[cU?R[5(dM'U1RU>U?_M9 nKn>n?[U4[WB5S5Hun>nWJURjnLWKULn?U?R[R:Xa[RnMU?RnIOIU?R[R:Xa[RnMWLnIO[SU?R35eWBRo[WM[URjS9WI55 M [U35[U,5:Xde[WAWBS9n([U,U35HunNnO[cUNRdS[GR5(dWNRdSWORdS'WORU(GR;dWORU(GR;dMWNRdR{5HunPnQUQWORdUP'M M U'U(URP4$s sn&n%f!,(df  G N=f!,(df  G N=f!,(df  G N=f!,(df  G N=f!,(df  G N=fs snfs snfs sn.n*f!,(df  GN=f!,(df  GN=f!,(df  GN=f!,(df  GN=fs snfs sn7f!,(df  GN=fs sn?n>fs sn?n>fs sn*n>fs snfs sn?n>f)Nr)_materialize_and_lift_constants)aot_export_module)2_disable_custom_triton_op_functional_decomposition_export_to_aten_ir_ignore_backend_decomps_verify_nn_module_stack_verify_placeholder_names_verify_stack_trace)ShapeEnvcHUSL=(d URRSL$rN)graph_signaturebackward_signature)eprs rU_is_joint_ir_decompO_decompose_and_get_gm_with_new_signature_constants.._is_joint_ir_decompps) D ( A!!44D@ rXF)remove_duplicate)!unwrap_tensor_subclass_parametersT) shape_envexport)&_enable_graph_inputs_of_type_nn_module_fakify_script_objects placeholdervalr4)r_check_autograd_state_prettify_placeholder_namesdecompose_custom_triton_ops)decompositions trace_jointoutput_loss_indexc[U[5(aU$[U[5(a[URS9$[U[5(a[ URS9$[U[ 5(a[ URS9$[U[ 5(a[ URS9$[S[U535e)NrOzType of old_arg not supported: ) isinstancer6r@rOr?r>r= RuntimeErrortype)old_argnew_phs rU update_argF_decompose_and_get_gm_with_new_signature_constants..update_argbs g/ 0 0N  0 0!v{{3 3  0 0!v{{3 3 !1 2 2#5 5  1 1" 4 4  &DII==++77AA ==u -udXJ?H+       ((:(:BLL#N%C  2 2 I I% ! 5YD&DS&I#  IIOODww-'dii.0ABB&*Oyy'008*,,,tyy7G7L7L*M*.))E*:*C*C*L*L"))/:"))$))E*:;$  , .0S 1 1    (.'<'< ( $(-a0-a0779' '9!#'+!4*/050K ($&%9$B$B$H$H$J 2 %KS %c+;<<)- ! %K 2$.*,,&:&>&># L+-F$ :/5H 'V&&(;'#<+-D ('!#'{ 1 . b #B!"&9:"C ##  0557GD!11&' d#8.335GD!2==(%a););<<<<r}}$77MM%%d+6& 55..444=8P4/??.>d5!.>I?-/__-J-J-LM-L'$-LM!&"!+I,5,= &&(9I '  ?( ! "I/R0*,/ OO . 0 <%$4$@ d  O $$&-0I " N*,T477m;STrxx~~&r*//2K  C(8$9 99 9.0@A&,kk1 fmB,<<+;R +;H< 77m # .xDIIN  $B'Y ==   ; ; 1 #2&  (I%%&9{S0()LRXX)V(WX LLN0454 k!nehhmm 4 4"&KN 6!!3!3!?!?@@GAt$(($45 0 '*///@!!3!3!?!?@AGAt  II txx!1!!4 5 KK OO  AL1:/:U:U0VW0VWQq0VW/??DDFGN K@U#nBU4VW4VDdii4VW WW W#2#H#H#N#N#P Z j )''33A6 9#7#79I9I"JJJJ)"2"22  & &// )"2"22   $$    " 4  $Q.R//<<=4 *+/?*? &&TYY488[^1Dq1H%IJ'++DKKE  >#11===#66OO ?../3r7I7I7U7U3VVVV%R%7%7%C%CD D4$((N3 1O ' ' *D 0D  !S->-@!ABGAttyy)F=DyyI///!77i222!88$';DII;%GHH   " 2 C&  C(8$9 99 9.l""24DE((--. ==#+==#7HMM%  OO2GG G"5"G"GG ++-1#$ a .F "BMM 11g29  1 1 . . I^@M-,00II " ":U=0$ "XWJ s5 {* {6z4 Az" 2z *z +A+z z" z4&{.{{**{*6{/.{4$|0' |3| ;<{: 7| ?||02} } !}3#} #&} (}A }$}* }0&&}5}5 z  zz" " z1 ,z44 { >{ { { {': | |  | | |- (|00 |? }rr9cURUR55 URRGHnURS:XdM[ R "UR5up4UHn[U[RR5(dM.URUR;dMJURS:XdM\UR[RR R"R$:XdMUR'URS5 URR)U5 M GM UR+5 SSS5 X4$!,(df  X4$=f)zI Removes redundant copy_ node that was introduced due to mutated buffer. output call_functionr4N)_set_replace_hookget_replace_hookrrrr` tree_flattenrjrrrrrObuffers_to_mutaterrrrdefaultrV erase_noder)rr9r/rjr=outs rUrrs   1BBD EHHNNDww(" --dii8C"3 66HH(;(M(MM FFo5 # eiinn.A.A.I.I I55chhqkBHH//4 #  F  ""! F E  ""s+)E( AE(E(;E( :E( AE(( E8c URUR55 UR5GH!n[U[R R 5(dM/0n0n[URR5HnURS:XaUR[R:XaURupxXGSU3n X;afURXY5 UH0n U R cMU R RXeU 5 M2 URR#U5 MXeU 'XU'MUR$XF'M GM$ SSS5 g!,(df  g=f)NrU.)rVrWmodulesrrr GraphModuler^rrrroperatorgetitemrjrVrgr[rO) rrmodule_call_graphrnode_idgetitemsr/rOidxnew_identrys rU _common_getitem_elimination_passri's#  o>>@ AjjlFfehh&:&:;;02G13HV\\//077o-$++AQAQ2Q"&))KF '0#7F)2283CD%6E$: % E E$(6*:!"&7  //5+/((. $(IIGM!1 # B A AsCE 7A E  E.old_module_call_graphc[R"U5n0nURRHCnURR S/5=n(dM(UR X5SR 'ME UH`nURnUcM/URQURQH.nUR UR UR 5UlM0 Mb U$)N from_noder) rdeepcopyrrrrrOrgrHrI) rrjnew_module_call_graph provenancer/historyrhrgxs rU_get_updated_module_call_graphrrDs!MM*?@"$JiimmK4 47 4+/99Jr{'' ( 'OO   89##8i&7&78A^^AFFAFF3AF9 ' ! rXrc :[UUUUUS9unnn[UUR5nURR UR R5 [ UUR5n [UURUUU UURURS9n U $)Nrrrr)rootrrrrange_constraintsrcrr) rRrrrcrrr_get_updated_range_constraintsrvrBrrr) rrrrrrr9rrnnew_range_constraintsexported_programs rU_decompose_exported_programrz\s ; #/)$?    ;  GGNN2??''(:   ' hh+//((,,  rXc\rSrSrSrSFSS.S\\RR\ \ \ 44S\RRS\S\ \ \\R\RR 44S S S \\S \\\\ S 4\ \ \ 44S\\ \ \\R\\R,R.44S\\\\4Sjjjr\\"SS9S55r\R<\"SS9S55r\\"SS9S55r\R<\"SS9S55r\\"SS9S55r \ R<\"SS9S55r \\"SS9S55r!\!R<\"SS9S55r!\"SS9S\"\RR 4Sj5r#\"SS9S\"\\ \RR 44Sj5r$\"SS9S\"\R4Sj5r%\"SS9S\"\\ \R44Sj5r&\\"SS9S 55r'\'R<\"SS9S!55r'\\"SS9S"55r(\(R<\"SS9S#55r(\\"SS9S$55r)\)R<\"SS9S%55r)\\"SS9S&55r*\*R<\"SS9S'55r*\\"SS9S\ 4S(j55r+\+R<\"SS9S)55r+\\"SS9S\ 4S*j55r,\,R<\"SS9S+55r,\\"SS9S,55r-\-R<\"SS9S-55r-\\"SS9S.55r.\.R<\"SS9S/55r.\\"SS9S055r/\/R<\"SS9S155r/S2r0S3\ S4\ S\ 4S5jr1S3\ S4\ S\ 4S6jr2S7r3S\ 4S8jr4S\RR4S9jr5S:r6\7SGS;\\ \RpRr\:4S<\;SS4S=jj5r\=SS4S?jr>S@r?\"SS9SA5r@\ASB5rBSSSSC.SHSDjjrCSErDg)IrBia Package of a program from :func:`export`. It contains an :class:`torch.fx.Graph` that represents Tensor computation, a state_dict containing tensor values of all lifted parameters and buffers, and various metadata. You can call an ExportedProgram like the original callable traced by :func:`export` with the same calling convention. To perform transformations on the graph, use ``.module`` property to access an :class:`torch.fx.GraphModule`. You can then use `FX transformation `_ to rewrite the graph. Afterwards, you can simply use :func:`export` again to construct a correct ExportedProgram. N) verifiersrurrrrvdict[sympy.Symbol, Any]rcr.rr|c >[RRR5Ul[ X5Ul[U[RR5(a/UR RRUR5 [UR X65 X0l X@l XPlUceX`lXplU=(d 0UlU =(d [$/n ['SU 55(deXlUR+5 g)Nc3B# UHn[U[5v M g7frN)rxr().0rQs rU +ExportedProgram.__init__..s>Iq:a**Is)rrrCodeGenr_create_graph_module_for_export _graph_modulerr`rrri_graph_signature _state_dict_range_constraints_module_call_graph_example_inputs _constantsr(all _verifiersvalidate) rPrurrrrvrcrrr|s rU__init__ExportedProgram.__init__s //1I>>>>># rXF)is_backward_compatiblecUR$rN)rrPs rUrExportedProgram.graph_modules!!!rXc[S5e)Nz7Unable to set ExportedProgram's graph_module attribute.rrPvalues rUrrsTUUrXc.URR$rN)rrrs rUrExportedProgram.graphs  &&&rXc[S5e)Nz0Unable to set ExportedProgram's graph attribute.rrs rUrrsMNNrXcUR$rN)rrs rUrExportedProgram.graph_signatures$$$rXc[S5e)Nz:Unable to set ExportedProgram's graph_signature attribute.rrs rUrrsWXXrXcUR$rN)rrs rUrExportedProgram.state_dictsrXc[S5e)Nz5Unable to set ExportedProgram's state_dict attribute.rrs rUrrsRSSrXrvc#F# UR5H upUv M g7f)z8 Returns an iterator over original module's parameters. N)r)rPr=params rU parametersExportedProgram.parameterss! --/HAK0!c#j# URRHnXRU4v M g7f)z Returns an iterator over original module parameters, yielding both the name of the parameter as well as the parameter itself. N)rrr)rP param_names rUr ExportedProgram.named_parameterss/ ..99Jooj99 9:s13c#F# UR5H upUv M g7f)z3 Returns an iterator over original module buffers. N)r)rPr=bufs rUbuffersExportedProgram.bufferss! ((*FAI+rc## [URR5nURRH.nX!;aX RU4v MX R U4v M0 g7f)zv Returns an iterator over original module buffers, yielding both the name of the buffer as well as the buffer itself. N)rrrrrr)rPr buffer_names rUrExportedProgram.named_bufferss^ "%T%9%9%P%P!Q//77K4!>>+#>>>!??;#??? 8sA)A+cUR$rN)rrs rUrv!ExportedProgram.range_constraints&&&rXc[S5e)Nz.s<(;1A$(;s)rrJr.r`tree_flatten_with_path_check_input_constraintsr)rPrjrkrJflat_args_with_path received_spec flat_argss rU_get_flat_args_with_check)ExportedProgram._get_flat_args_with_checksg..((  #F4F-3-J-J N. * %%&9:<(;<< ''rXrjrkcURRnURX5upEUb&[XS[5(d[ SUSU35e/nUR RGH nUR[R:XaM$UR[R[R4;acURSLa*URURUR 5 MURUR"UR 5 MUR[R$[R&4;dMURURUR 5 GM [)U5nXd-$)aTransform args, kwargs of __call__ to args for graph_module. self.graph_module takes stuff from state dict as inputs. The invariant is for ep: ExportedProgram is ep(args, kwargs) == ep.postprocess(ep.graph_module(ep.graph_module_flat_inputs(args, kwargs))) z>Trying to flatten user inputs with exported input tree spec: z- but actually got inputs with tree spec of: F)rrJrr-rzrrrrr9rrrrrrrrCONSTANT_TENSOR CUSTOM_OBJr)rPrjrkrJrradditional_inputsinput_s rU_graph_module_flat_inputs)ExportedProgram._graph_module_flat_inputss\..((#'#A#A$#O  } $A( ( Q)@ /# **66F{{i222##  !$$-&,,T^^FMM-JK%,,T__V]]-KL))$$!"(( )FG#7$""34!,,rXc[S5e)Nz\Unable to call ExportedProgram directly. You should use `exported_program.module()` instead.r)rPrjrks rU__call__ExportedProgram.__call__s B  rXc r^SSKJs Jn URX#5upVURR GbUR RnUR Rn[U5[U5-n USU n UR Rn U b'[[U R555n USU nXSn[R"XRR 5nUR R&Vs/sH%nUR([*R,:XdM#UPM' nn[/U 5HunnUR R0UmTR([2R4:Xa*TR6ceUUR8TR6'MgTR([2R::XaCTR6ce[U4Sj[/U555nUUR=U5 M[?STR(35e U$U$![ aD [R""U5upmUR%SURR SU 35ef=fs snf!UR R&Vs/sH%nUR([*R,:XdM#UPM' Os snfnn[/U 5HunnUR R0UmTR([2R4:Xa*TR6ceUUR8TR6'MgTR([2R::XaCTR6ce[U4Sj[/U555nUUR=U5 M[?STR(35e f=f)zProcess potential mutations to the input. Because self.graph_module is functional, so mutations has to be written back after execution of graph_module. rNz@Trying to flatten user outputs with exported output tree spec: z. but actually got outputs with tree spec of: c3x># UH/upURRTR:XdM+Uv M1 g7frN)rrOr)rrSrE output_specs rUrDExportedProgram._postprocess_graph_module_outputs.. s2%+A#xx}} 0B0BBA+As*: :zUnexpected kind: ) torch._export.error_exporterrorrrrKrrYrr assertion_dep_tokennextiterrr`r ExceptionrX InternalErrorrrr9rrrr;rrrrcopy_r")rPresr orig_kwargsrrr=buffer_mutationuser_input_mutation num_mutatedmutated_valuesrassertion_dep_token_indexrrErrSrindexrs @rU!_postprocess_graph_module_outputs1ExportedProgram._postprocess_graph_module_outputsst ,+55iM >> " " ."22DDO"&"6"6"L"L o.5H1IIK +.N#'"6"6"J"J ".,06I6N6N6P1Q,R)445l#C U++C1H1HI!% 4 4 @ @ @yyI$8$88 @ !*. 9HAu"&"6"6"C"CA"FK"'':+E+EE*11===>C (:(:;$))Z-K-KK*11=== $%+4[+A%! "%(..u5,/@AQAQ@R-STT!: s ; #)#6#6s#; ))W~~../0E$o'  $ 4 4 @ @ @yyI$8$88 @  !*. 9HAu"&"6"6"C"CA"FK"'':+E+EE*11===>C (:(:;$))Z-K-KK*11=== $%+4[+A%! "%(..u5,/@AQAQ@R-STT!:s8*H%"I6,I6%AI33I;;N6"K:KC6N6cURRSSS9RSS5nSUSURSURS3nU$)NF) print_outputcolored z zExportedProgram: z Graph signature: z Range constraints: )rprint_readablereplacerrv)rPrstrings rU__str__ExportedProgram.__str__sp((778 '$ !  .! $ 4 456""&"8"8!9 =   rXcSSKJn U"U5nSS[4SjjnSS[4Sjjn[R"X25Ul[R"XB5UlU$)zO Returns a self contained GraphModule with all the parameters/buffers inlined. r4)&_unlift_exported_program_lifted_statesmodec[S5e)Nz%Calling train() is not supported yet.NotImplementedErrorrPrs rU_train&ExportedProgram.module.._train*s%&MN NrXc[S5e)Nz$Calling eval() is not supported yet.rrs rU_eval%ExportedProgram.module.._eval-s%&LM MrXT)rrbooltypes MethodTypetraineval)rPrrrrs rUrExportedProgram.module"sY D7= Ot O Nd N''7 &&u5  rXc[S[URR55[ URR55$)Nc3j# UH)upUR[R:XdM%Uv M+ g7frN)rr9r)rrSss rUr=ExportedProgram._num_lifted_params_buffers..6s- HDA66Y111Hs$3 3)rrrrr rs rU_num_lifted_params_buffers*ExportedProgram._num_lifted_params_buffers4sC %d&;&;&G&GH  %%11 2   rXrrcUc [5O [U5n[U[5(aUR 5n[ U5unn[ UUUSUS9$)a Run a set of decompositions on the exported program and returns a new exported program. By default we will run the Core ATen decompositions to get operators in the `Core ATen Operator Set `_. For now, we do not decompose joint graphs. Args: decomp_table: An optional argument that specifies decomp behaviour for Aten ops (1) If None, we decompose to core aten decompositions (2) If empty, we don't decompose any operator Some examples: If you don't want to decompose anything .. code-block:: python ep = torch.export.export(model, ...) ep = ep.run_decompositions(decomp_table={}) If you want to get a core aten operator set except for certain operator, you can do following: .. code-block:: python ep = torch.export.export(model, ...) decomp_table = torch.export.default_decompositions() decomp_table[your_op] = your_custom_decomp ep = ep.run_decompositions(decomp_table=decomp_table) Nrt)rEryrr0 materializerrz)rPrr _decomp_tablerrs rUrun_decompositions"ExportedProgram.run_decompositions>snP)5(< " $$|BT  m%6 7 7)557M" 9 G  + ' 3!(C   rXpassesc j[[U55nSSKJn U"5 U"UR5nSSS5 Wb URO URnUceXPRLaUR (dU$S[ S[RRS[ 4Sjn[UURU"URU5UR[UUR5[ R""UR$5UR&UR(UR*S9 nURR,R/URR,5 URR,R/URR,5 U$!,(df  GNu=f)Nr)r old_signaturenew_gmrvc &/n[URR5Hup4URS:wa OU[ UR 5:dS5eUR Un[ UR[[45(a URO$[UR5"UR5nUR[URUURUR 55 M [#URR5SnURS:Xde/n[UR$S5Hup4U[ UR&5:dS5eUR&Un [ U R[[45(a U RO$[U R5"UR5nUR[)U RXiR55 M [+X(S9n U $)z7 Update the graph signature's user_input/user_outputs. rz-Number of inputs changed after transformationrrTrz.Number of outputs changed after transformationr)rrrrr rrrr6r7rrOrr:rrrr^rjrr<r8) r1r2new_input_specsrSr/old_input_specr output_nodenew_output_specsold_output_spec new_signatures rU_get_updated_graph_signatureKExportedProgram._transform_do_not_use.._get_updated_graph_signatures!O$V\\%7%7877m+3!--CBC"/!:!:1!="&**-=?P,Q#&&n001$))<  &&&++&--&11 90v||11226K>>X- --! $[%5%5a%893!..DCD#0"<"@Q-R$''o112499= !''33S:P:PQ: 1+M! rX rurrrrvrcrrr|)r3r^rrrmodifiedr8rrr`rBrrrrwrvrrmrrrr|rr)rPr/pmrrtransformed_gmr:transformed_eps rU_transform_do_not_use%ExportedProgram._transform_do_not_usesf f & @ $ &T&&'C'-0_))$BSBS))) .. .s||K7 !/7 !HH((7 !"7 !r) &&8$$n<&&#mmD,C,CD..nnnn  ##((//0A0A0F0FG##((//0@0@0E0EFi' &s F## F2c`SSKJn URRVs/sHo3RS:XdMUPM nn[ X@R R5VVs/sH'up5UR[R:XdM%UPM) nnnU"XaUR5 gs snfs snnf)Nr)"_check_input_constraints_for_graphr) torch._export.utilsrDrrrr rrrr9rrv)rPrrDr: placeholdersr&input_placeholderss rUr(ExportedProgram._check_input_constraintssJ#'::#3#3M#3att}7L#3 ML*>*>*J*JK Kvv--- K  + T5K5K N sB%B% $B*B*c$UR5 grN) _validaters rUrExportedProgram.validates  rXc[UR5S:dS5eURHnU"5RU5 M g)Nrz0ExportedProgram must have at least one verifier.)r r|check)rPrQs rUrJExportedProgram._validates@  ! # > = > #A CIIdO rX)rrr|c "[UURUUbUO UR[R"UR 5[R"UR 5URUbUO URUbUS9 $URS9 $)Nr<) rBrrrrmrvrrrr|)rPrrrrr|s rU_updateExportedProgram._updates$$+%/%;z"mmD,B,BC"mmD,C,CD..#,#8idnn#,#8i  ?Cnn  rX)rrrrrrrr)NN)NF)rvrB)ErZr[r\r]__doc__r rrModuleryrbrrGraphr8r#rr^rCr rr r ScriptObjectrr(rpropertyr1rsetterrrrrrrrrrvrcrrrrr|rrrrrrrrr(rq_ops OperatorBaserrr-PassTyperArrrrJrPrcrYrXrUrBrBs .LP %59%EHHOOT#s(^34%xx~~%. % eELL%((2D2D$DEEF % 5 % 0%!uS#XS#X'F!GH% eELL*:EHHr?r@rA__all__rr`rZ dataclassrDrCrqrContextrrzrr AutogradCPU AutogradCUDA AutogradMeta AutogradXLA AutogradLazy AutogradIPU AutogradXPU AutogradMPS AutogradHPUAutogradPrivateUse1AutogradPrivateUse2AutogradPrivateUse3rCPUCUDAMetaXLALazyIPUXPUMPSHPUrrrryrXrYrrrEintrRrrir^rrrzrBrrwrrYrXrUrs  "$%GGB? 9?6IJ ; $$",*@B716:"  UXX))*HZ,@@ A  ' ' '  444 GG>>GG>> G  G: HH$$ HH%% HH%% HH$$ HH%% HH$$ HH$$ HH$$ HH$$ HH,, HH,, HH,, ,(, HH HH HH HH HH HH HH HH HH X$X$v  )-uzz..89)- 4 ''1 2C 78)-Xo2 //9:o2ejj55x?@ o2 sm o2d # #3G# 588  !5 56#2. .:! !0!01 //9:1ejj55x?@ 1 sm 1 "& 1h    D $$BF >.rX