i %SSKJr SSKJrJrJrJrJr SSK r SSK J r \\ R\ \\\\\ R$\ R&\ R(\ R*S4 r0r\\\ \ 4\\ R0R2\ R0R444\S'0r\\\ \ 4\4\S'S\ R:R<S\ R:R>S \S \\ \ R@RB\ R:R>44S jr"S\ R:R<S \#\ \$\4S \\\ \ R:R<\ R:R>44S jr%S\ R:R<S \\\ \ R:R<\ R:R>44Sjr&S\ R:R<S \\\ \ R:R<\ R:R>44Sjr'S\ R:R<S \\\ \ R:R<\ R:R>44Sjr(S\ R:R<S\\ R:R>/S4S S4Sjr)g)) FunctionType)CallableDictListTupleUnionN)HigherOrderOperator_cache_ops_dict_cache_fake_ops_dict graph_modulenode arg_indexreturnc.URUn[U[RR5(deUR S:Xde[UR [5(deURUR 5nUR XA4$)Nget_attr) args isinstancetorchfxNodeoptargetstr get_submodule)r r r submod_node submodules [/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/exir/graph_module.py_get_submoduler's))I&K k588== 1 11 1 >>Z '' ' k((# . .. .**;+=+=>I   y ..op_to_submodule_arg_indexc /nURRHUnURS:waMUH:nURULaMXHnUR [ XU55 M M< MW U$)a@ Returns a list of submodules used for control flow operations that are in the given toplevel graph (does not look into submodules). Specifically, the returned value is a list containing tuples of (name of the submodule that's stored in the graph module, the submodule itself, and the fx node that uses this submodule). call_function)graphnodesrrappendr)r r control_flow_submodulesr ris r_get_control_flow_submodulesr(3ss!""(( 77o % +B{{"$.2'..~lRS/TU3, ) #"rc [U[RRRSS/[RRR S/[RRR S/05$)ag Returns a list of submodules used for control flow operations (torch.ops.higher_order.cond/map/scan) that are in the given toplevel graph (does not look into submodules). Specifically, the returned value is a list containing tuples of (name of the submodule that's stored in the graph module, the submodule itself, and the fx node that uses this submodule). r)r(rops higher_ordercondmap_implscanr s rget_control_flow_submodulesr2Ls` ( II " " ' '!Q II " " + +aS II " " ' '!  rc[U[RRRSS/[RRR SS/05$)ai Returns a list of submodules used for control flow operations (torch.ops.higher_order.cond/while_loop) that are in the given toplevel graph (does not look into submodules). Specifically, the returned value is a list containing tuples of (name of the submodule that's stored in the graph module, the submodule itself, and the fx node that uses this submodule). r*r+r)r(rr,r-r. while_loopr1s rget_cond_while_submodulesr5`sK ( II " " ' '!Q II " " - -1v  rcd[U[RRRS/05$)a Returns a list of submodules used for scan operations (torch.ops.higher_order.scan) that are in the given toplevel graph (does not look into submodules). Specifically, the returned value is a list containing tuples of (name of the submodule that's stored in the graph module, the submodule itself, and the fx node that uses this submodule). For scan, the combine_fn submodule is at argument index 0. The scan operator signature is: scan(combine_fn, init, xs, additional_inputs) r)r(rr,r-r0r1s rget_scan_submodulesr7ss0 ( II " " ' '!  rgmnode_opc~[U[RR5(dS[ U535eU/nU(aqUR S5nUR RH nU"U5 M [U5VVs/sHupVnUPM nnnURU5 U(aMpggs snnf)z9Traverse the graph module and apply node_op to each node.zExpected GraphModule, got rN) rrr GraphModuletypepopr#r$r2extend)r8r9queuecurrent_graph_moduler _rr&s rbfs_trace_with_node_processrBs b%((.. / /X3MdSUhZ1XX / DE $yy|(..44D DM5 $??S#T# #Ta #T #   ,- % # s B9)*typesrfunctiontypingrrrrrr torch._opsr Tensorrintfloatboolcomplexdtypedevice memory_formatlayout LeafValuer _ops OpOverloadOpOverloadPacket__annotations__r rr;rnnModulerdictlistr(r2r5r7rBrrrZsp+55 *  LL   KK LL  LL    $ #s(OU5::00%**2M2MMNN9;d5c?H45: /((&& /.3hhmm /HK / 3 ./ /#((&&##$7c$BC# %UXX))588==8 9:#2((&& %UXX))588==8 9:(((&& %UXX))588==8 9:&((&& %UXX))588==8 9:*. .'/0E'F. .r