BVhdZddlZddlmZddlmZddlmZddlmZddlmZddl m Z Gd d e Z d a Gd d e ZGddejZdZy)zHandles function calls, by generating compiled function names and calls. Note: this transformer does not rename the top level object being converted; that is the caller's responsibility. Requires function_scopes. N) converter)anno)parser) qual_names) templates) ag_loggingceZdZdZdZy) _FunctionTcd|_yN) context_nameselfs a/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/autograph/converters/call_trees.py__init__z_Function.__init__(s DN)__name__ __module__ __qualname__no_rootrrrr r $s 'rr Fc4eZdZdZdZdZdZdZdZdZ y) _ArgTemplateBuilderzConstructs a tuple representing the positional arguments in a call. Example (yes, it's legal Python 3): f(*args1, b, *args2, c, d) -> args1 + (b,) + args2 + (c, d) c.g|_g|_d|_y)NF)_arg_accumulator_argspec _finalizedrs rrz_ArgTemplateBuilder.__init__7sDDMDOrc|jrT|jjtj|jtj g|_yy)N)eltsctx)rrappendgastTupleLoadrs r _consume_argsz!_ArgTemplateBuilder._consume_args<sE  mm **$//TYY[ AC drc:|jj|yr )rr!ras radd_argz_ArgTemplateBuilder.add_argBs  #rc |j|jjtjtj dtj dd|gdy)Ntupler annotation type_commentrargskeywords)r%rr!r"CallNamer$r's r add_starargz_ArgTemplateBuilder.add_starargEsPMM IITYY[T N  rc2|jd|_y)NT)r%rrs rfinalizez_ArgTemplateBuilder.finalizeNsDOrc\|jsJ|jrk|jd}tdt|jD]8}t j |t j |j|}:|St jgt jS)Nr) rrrangelenr"BinOpAddr#r$)rresultis rto_astz_ArgTemplateBuilder.to_astRs ??? }}}}QfQDMM*+B!FDHHJ a0@AB m ::b$))+ &&rN) rrr__doc__rr%r)r4r6r?rrrrr/s% ! $'rrc4eZdZdZdZdZdZdZdZdZ y) CallTreeTransformerz9Transforms the call tree by renaming transformed symbols.ctj|ds|j|S|jt5}tj |d|_|j|cdddS#1swYyxYw)Nfunction_context_name)rhasanno generic_visitstater getannor )rnodefn_scopes r visit_Lambdaz CallTreeTransformer.visit_Lambda_si <<5 6   %% I &("ll41HIh    %&&&s ,A11A:c|j|j|_|j|jj|j_t |jj D]0\}}| |j ||jj |<2|jt5}tj|dsJdtj|d|_ |j|j|_ |jr |j |j|_|cdddS#1swYyxYw)NrDzCThe function_scopes converter always creates a scope for functions.) visit_blockdecorator_listr0defaults enumerate kw_defaultsvisitrGr rrErHr bodyreturns)rrIr>drJs rvisit_FunctionDefz%CallTreeTransformer.visit_FunctionDefhs**4+>+>?D))$))*<*<=DII$))//011 #'::a= a 1 I  (\\$ 7 8Q OQ 8"ll41HIh""499-di zz$,,/    s :BE  EcF|j|j|_|Sr )rMrSrrIs r visit_WithzCallTreeTransformer.visit_Withzs  +DI Krct}|jD]I}t|tjr|j |j 9|j|K|j|jS)zBTies together all positional and *arg arguments in a single tuple.) rr0 isinstancer"Starredr4valuer)r6r?)rrIbuilderr(s r_args_to_tuplez"CallTreeTransformer._args_to_tuplesf"#G YY At|| $AGG$    >> rc|jrKtjtjdtjddd|jSt j dS)zATies together all keyword and **kwarg arguments in a single dict.dictNr,rr/None)r1r"r2r3r$rparse_expressionrXs r_kwargs_to_dictz#CallTreeTransformer._kwargs_to_dictsS }} YY ))$))+$T K== ""  $ $V ,,rcttj|jtjj d}|j tj}|j|}|jdr|S|j|dzr|S|dvrtstjdda |S|dk(rI|jjj j#t$j&j(s|Sd }t+j,||j|j/||j1|| }|S) N)defaultzag__..)z pdb.set_tracezipdb.set_trace breakpointzDetected `pdb.set_trace()` in user code. The code generated by AutoGraph is not optimized for step-by-step debugging. See https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/autograph/g3doc/reference/debugging.md.TprintzA ag__.converted_call(func, args, kwargs, function_ctx) )funcr0kwargs function_ctx)strrrHrkBasicQNrGr r rF startswithset_trace_warnedrwarningr useroptionsusesrFeatureBUILTIN_FUNCTIONSrreplace_as_expressionr_rd)rrI full_namerDtemplatenew_calls r visit_CallzCallTreeTransformer.visit_Calls&DLLDJJMM2FGI JJy1>>   d #D G$ k1C78 k EE      kW HHMM ! ! & &y'8'8'J'J K kH.. YY   &##D)* ,H OrN) rrrr@rKrVrYr_rdr}rrrrBrB\s#A&$ $ -0rrBcdtj|}t|j|}|S)zTransform function call to the compiled counterparts. Args: node: AST ctx: EntityContext Returns: A tuple (node, new_names): node: The transformed AST new_names: set(string), containing any newly-generated names )rresolverBrR)rIr s r transformrs.   D !$ S ! ' ' -$ +r)r@r" tensorflow.python.autograph.corer tensorflow.python.autograph.pyctrrrr!tensorflow.python.autograph.utilsrobjectr rrrBaserBrrrrrsZ 613768 *'&*'Zp)..pfr