flashinfer.kda_training.recurrent_kda_training_backward

flashinfer.kda_training.recurrent_kda_training_backward(context: RecurrentKDATrainingContext, do: Tensor, dfinal_state: Tensor, out: Sequence[Tensor] | None = None) → tuple[Tensor, ...]

Differentiate a saved route context without rerunning forward recurrence.