flashinfer.cake\_rmsnorm\_silu\_gate\_train.CakeRMSNormSiLUGateFunction
=======================================================================

.. currentmodule:: flashinfer.cake_rmsnorm_silu_gate_train

.. autoclass:: CakeRMSNormSiLUGateFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~CakeRMSNormSiLUGateFunction.__init__
      ~CakeRMSNormSiLUGateFunction.apply
      ~CakeRMSNormSiLUGateFunction.backward
      ~CakeRMSNormSiLUGateFunction.forward
      ~CakeRMSNormSiLUGateFunction.jvp
      ~CakeRMSNormSiLUGateFunction.mark_dirty
      ~CakeRMSNormSiLUGateFunction.mark_non_differentiable
      ~CakeRMSNormSiLUGateFunction.mark_shared_storage
      ~CakeRMSNormSiLUGateFunction.maybe_clear_saved_tensors
      ~CakeRMSNormSiLUGateFunction.name
      ~CakeRMSNormSiLUGateFunction.register_hook
      ~CakeRMSNormSiLUGateFunction.register_prehook
      ~CakeRMSNormSiLUGateFunction.save_for_backward
      ~CakeRMSNormSiLUGateFunction.save_for_forward
      ~CakeRMSNormSiLUGateFunction.set_materialize_grads
      ~CakeRMSNormSiLUGateFunction.set_output_grad_dtype
      ~CakeRMSNormSiLUGateFunction.setup_context
      ~CakeRMSNormSiLUGateFunction.vjp
      ~CakeRMSNormSiLUGateFunction.vmap
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~CakeRMSNormSiLUGateFunction.boxed_grads_call
      ~CakeRMSNormSiLUGateFunction.clear_saved_tensors_on_access
      ~CakeRMSNormSiLUGateFunction.dirty_tensors
      ~CakeRMSNormSiLUGateFunction.generate_vmap_rule
      ~CakeRMSNormSiLUGateFunction.materialize_grads
      ~CakeRMSNormSiLUGateFunction.metadata
      ~CakeRMSNormSiLUGateFunction.needs_input_grad
      ~CakeRMSNormSiLUGateFunction.next_functions
      ~CakeRMSNormSiLUGateFunction.non_differentiable
      ~CakeRMSNormSiLUGateFunction.output_grad_dtypes
      ~CakeRMSNormSiLUGateFunction.requires_grad
      ~CakeRMSNormSiLUGateFunction.saved_for_forward
      ~CakeRMSNormSiLUGateFunction.saved_tensors
      ~CakeRMSNormSiLUGateFunction.saved_variables
      ~CakeRMSNormSiLUGateFunction.to_save