flashinfer.cake\_mhc\_det\_train.CakeMHCDetTrainFinishFunction ============================================================== .. currentmodule:: flashinfer.cake_mhc_det_train .. autoclass:: CakeMHCDetTrainFinishFunction .. automethod:: __init__ .. rubric:: Methods .. autosummary:: ~CakeMHCDetTrainFinishFunction.__init__ ~CakeMHCDetTrainFinishFunction.apply ~CakeMHCDetTrainFinishFunction.backward ~CakeMHCDetTrainFinishFunction.forward ~CakeMHCDetTrainFinishFunction.jvp ~CakeMHCDetTrainFinishFunction.mark_dirty ~CakeMHCDetTrainFinishFunction.mark_non_differentiable ~CakeMHCDetTrainFinishFunction.mark_shared_storage ~CakeMHCDetTrainFinishFunction.maybe_clear_saved_tensors ~CakeMHCDetTrainFinishFunction.name ~CakeMHCDetTrainFinishFunction.register_hook ~CakeMHCDetTrainFinishFunction.register_prehook ~CakeMHCDetTrainFinishFunction.save_for_backward ~CakeMHCDetTrainFinishFunction.save_for_forward ~CakeMHCDetTrainFinishFunction.set_materialize_grads ~CakeMHCDetTrainFinishFunction.set_output_grad_dtype ~CakeMHCDetTrainFinishFunction.setup_context ~CakeMHCDetTrainFinishFunction.vjp ~CakeMHCDetTrainFinishFunction.vmap .. rubric:: Attributes .. autosummary:: ~CakeMHCDetTrainFinishFunction.boxed_grads_call ~CakeMHCDetTrainFinishFunction.clear_saved_tensors_on_access ~CakeMHCDetTrainFinishFunction.dirty_tensors ~CakeMHCDetTrainFinishFunction.generate_vmap_rule ~CakeMHCDetTrainFinishFunction.materialize_grads ~CakeMHCDetTrainFinishFunction.metadata ~CakeMHCDetTrainFinishFunction.needs_input_grad ~CakeMHCDetTrainFinishFunction.next_functions ~CakeMHCDetTrainFinishFunction.non_differentiable ~CakeMHCDetTrainFinishFunction.output_grad_dtypes ~CakeMHCDetTrainFinishFunction.requires_grad ~CakeMHCDetTrainFinishFunction.saved_for_forward ~CakeMHCDetTrainFinishFunction.saved_tensors ~CakeMHCDetTrainFinishFunction.saved_variables ~CakeMHCDetTrainFinishFunction.to_save