oa::ml::matrix::mamba3_preprocess_backward

Compute the packed projection and dt-bias adjoints for Mamba-3 preprocess. output_gradients follows the exact eight shapes returned by mamba3_preprocess. Its dt-bias reduction is ordered by increasing row. Errors Returns an error for invalid configuration, incompatible gradients, shape, dtype or Engine ownership, arithmetic overflow, or failed runtime recording.

FnRust public API

Function Documentation

pub fn mamba3_preprocess_backward( projected: &Matrix, dt_bias: &Matrix, output_gradients: &Mamba3PreprocessResult, config: Mamba3PreprocessConfig ) -> Result<Mamba3PreprocessBackward>
Compute the packed projection and dt-bias adjoints for Mamba-3 preprocess. output_gradients follows the exact eight shapes returned by mamba3_preprocess. Its dt-bias reduction is ordered by increasing row. Errors Returns an error for invalid configuration, incompatible gradients, shape, dtype or Engine ownership, arithmetic overflow, or failed runtime recording.