flashinfer.cake\_rmsnorm\_det\_train.CakeRMSNormDetFunction
===========================================================

.. currentmodule:: flashinfer.cake_rmsnorm_det_train

.. autoclass:: CakeRMSNormDetFunction

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

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