
    WhC                     T   d dl mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZFmGZGmHZHmIZImJZJmKZKmLZL ddddZEd dlMmNZN dS )    )KAbstractedAxesSpecAbstractedAxisNameBoundedAxisSizeConstConstFoldRuleConstVarDCERule	DebugInfoDynamicJaxprTraceDynamicJaxprTracerForwardingRuleFreeVarJaxprEqnRecipeJaxprStackFrame
JaxprTraceJaxprTracerJaxprTracerRecipeLambdaBindingParamsUpdaterPartialEvalCustomResultPartialEvalCustomRule
PartialValResAvalUpdaterTracerAsNameTracerIdValabstract_eval_funarg_info_allcall_padding_rulecall_param_updaterscall_partial_eval_custom_rulecall_partial_eval_rulesclose_jaxpr$closed_call_partial_eval_custom_ruleconfigconst_fold_rulesconvert_constvars_jaxprconvert_envvars_to_constvarsconvert_invars_to_constvarscustom_partial_eval_rulescustom_staging_rules	dce_jaxprdce_jaxpr_call_ruledce_jaxpr_closed_call_ruledce_jaxpr_consts	dce_rules
debug_infodebug_info_finaldef_trivial_paddingforwarding_ruleshas_effectsinfer_lambda_input_typeinstantiate_const_atmake_jaxpr_effectsmove_binders_to_backmove_binders_to_frontnew_eqn_recipe	pad_jaxprpadding_rulespartial_eval_jaxpr_custom.partial_eval_jaxpr_custom_rule_not_implementedpartial_eval_jaxpr_custom_rulespartial_eval_jaxpr_nounitspartial_eval_wrapper_nounitspartition_pvalsrecipe_to_eqnresult_infosig_infotrace_to_jaxpr_dynamictrace_to_jaxpr_dynamic2trace_to_jaxpr_nounitstrace_to_subjaxpr_nounitstrace_to_subjaxpr_nounits_fwdtracers_to_jaxprtrivial_ctxNkeep_inputsc                >    t          | |||          \  }}}\   |||fS )NrN   )_trace_to_jaxpr_dynamic)funin_avalsr1   rO   jaxpr	out_avalsconstss          c/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/jax/interpreters/partial_eval.pyrG   rG   _   s5    !8	8Z[": ": ":%FB		6	!!    )Jaxpr)N)O"jax._src.interpreters.partial_evalr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rQ   rH   rI   rJ   rK   rL   rM   jax._src.corerY    rX   rW   <module>r]      s  L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L L`"$ " " " " " ) ( ( ( ( ( ( (rX   