Skip to main content

mat_mul_in

Function mat_mul_in 

Source
pub fn mat_mul_in<T: BlasMul>(
    session: &mut dyn Engine,
    a: &Matrix<T>,
    b: &Matrix<T>,
) -> Result<Matrix<T>, MatrixMulError>
Expand description

Matrix multiplication A * B on a caller-held engine.

Same kernel, shape validation and result container as mat_mul, entered through the given session instead of the process-global one.

§Errors

Returns MatrixMulError when the shapes disagree or the engine operation fails.

§Examples

mat_mul_in is usually reached through explicit::Session::mat_mul, which holds the session for the whole stage.