o ?e@sdZddlZddlmZddlmZddlmZddlmZddlmZddl m Z Gd d d e Z d a Gd d d e ZGdddejZddZdS)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_loggingc@seZdZdZddZdS) _FunctionTcCs d|_dSN) context_nameselfr r/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/autograph/converters/call_trees.py__init__(s z_Function.__init__N)__name__ __module__ __qualname__Zno_rootrr r r rr$s rFc@@eZdZdZddZddZddZdd Zd d Zd d Z dS)_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) cCsg|_g|_d|_dS)NF)_arg_accumulator_argspec _finalizedr r r rr7s z_ArgTemplateBuilder.__init__cCs0|jr|jtj|jtdg|_dSdS)N)eltsctx)rrappendgastTupleLoadr r r r _consume_args<s  z!_ArgTemplateBuilder._consume_argscCs|j|dSr )rrr ar r radd_argBsz_ArgTemplateBuilder.add_argc Cs8||jtjtjdtddd|gdddS)Ntupler annotation type_commentr argskeywords)rrrrCallNamerrr r r add_starargEs z_ArgTemplateBuilder.add_starargcCs|d|_dS)NT)rrr r r rfinalizeNs z_ArgTemplateBuilder.finalizecCs\|jsJ|jr&|jd}tdt|jD]}t|t|j|}q|StgtS)Nr) rrrangelenrBinOpAddrr)r resultir r rto_astRs  z_ArgTemplateBuilder.to_astN) rrr__doc__rrr!r+r,r4r r r rr/s rc@r)CallTreeTransformerz9Transforms the call tree by renaming transformed symbols.cCs\t|ds ||S|jt}t|d|_||WdS1s'wYdS)Nfunction_context_name)rhasanno generic_visitstatergetannor )r nodefn_scoper r r visit_Lambda_s   $z CallTreeTransformer.visit_LambdacCs||j|_||jj|j_t|jjD]\}}|dur'|||jj|<q|jt,}t |ds8Jdt |d|_ ||j |_ |jrP||j|_|WdS1s\wYdS)Nr7zCThe function_scopes converter always creates a scope for functions.) visit_blockdecorator_listr'defaults enumerate kw_defaultsvisitr:rrr8r;r bodyreturns)r r<r3dr=r r rvisit_FunctionDefhs  $z%CallTreeTransformer.visit_FunctionDefcCs||j|_|Sr )r?rEr r<r r r visit_WithzszCallTreeTransformer.visit_WithcCsFt}|jD]}t|tjr||jq||q|| S)zBTies together all positional and *arg arguments in a single tuple.) rr' isinstancerStarredr+valuer!r,r4)r r<Zbuilderr r r r_args_to_tuples   z"CallTreeTransformer._args_to_tuplecCs4|jrtjtjdtdddd|jdStdS)zATies together all keyword and **kwarg arguments in a single dict.dictNr#r r&None)r(rr)r*rrZparse_expressionrIr r r_kwargs_to_dicts  z#CallTreeTransformer._kwargs_to_dictcCsttj|jtjjdd}|jtj}| |}| dr|S| |dr(|S|dvr7t s5t dda |S|dkrG|jjjtjjsG|Sd }tj||j|||||d }|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) )funcr'kwargsZ function_ctx)strrr;rWZBasicZQNr:rr r9 startswithset_trace_warnedrwarningruseroptionsZusesrZFeatureZBUILTIN_FUNCTIONSrZreplace_as_expressionrNrQ)r r<Z full_namer7templateZnew_callr r r visit_Calls6   zCallTreeTransformer.visit_CallN) rrrr5r>rHrJrNrQr`r r r rr6\s  r6cCst|}t||}|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 )rresolver6rD)r<rr r r transforms rb)r5rZ tensorflow.python.autograph.corerZ tensorflow.python.autograph.pyctrrrrZ!tensorflow.python.autograph.utilsrobjectrr[rZBaser6rbr r r rs      - s