jax.extend.lowering.LoweringRuleContext
=======================================

.. currentmodule:: jax.extend.lowering

.. autoclass:: LoweringRuleContext

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~LoweringRuleContext.__init__
      ~LoweringRuleContext.is_forward_compat
      ~LoweringRuleContext.replace
      ~LoweringRuleContext.set_tokens_out
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~LoweringRuleContext.axis_size_env
      ~LoweringRuleContext.dim_var_values
      ~LoweringRuleContext.jaxpr_eqn_ctx
      ~LoweringRuleContext.platforms
      ~LoweringRuleContext.module_context
      ~LoweringRuleContext.name_stack
      ~LoweringRuleContext.traceback
      ~LoweringRuleContext.primitive
      ~LoweringRuleContext.avals_in
      ~LoweringRuleContext.avals_out
      ~LoweringRuleContext.tokens_in
      ~LoweringRuleContext.tokens_out
      ~LoweringRuleContext.const_lowering