Struct CpuGemmRequest 

pub struct CpuGemmRequest<'request, 'input, 'output> { /* private fields */ }

Validated borrowed GEMM request.

A batch count of one is a single GEMM. A larger batch count is a strided batched GEMM.

Examples

use tenferro_cpu::provider::CpuGemmRequest;
assert!(request.batch_count() >= 1);

Implementations

impl<'request, 'input, 'output> CpuGemmRequest<'request, 'input, 'output>

pub fn lhs(&self) -> &TensorRead<'input>

Return the borrowed left input.

pub fn rhs(&self) -> &TensorRead<'input>

Return the borrowed right input.

pub fn output(&mut self) -> &mut TensorWrite<'output>

Reborrow the writable output for the duration of the current call.

pub fn rows(&self) -> usize

Return the number of output rows.

pub fn columns(&self) -> usize

Return the number of output columns.

pub fn contracted(&self) -> usize

Return the contracted dimension.

pub fn batch_count(&self) -> usize

Return the number of matrices in this strided batch.

pub fn lhs_layout(&self) -> CpuBatchedMatrixLayout

Return the left input matrix layout.

pub fn rhs_layout(&self) -> CpuBatchedMatrixLayout

Return the right input matrix layout.

pub fn output_layout(&self) -> CpuBatchedMatrixLayout

Return the output matrix layout.

pub fn accumulation(&self) -> DotGeneralAccumulation

Return conjugation and alpha/beta update semantics.

Trait Implementations

impl<'request, 'input, 'output> Debug for CpuGemmRequest<'request, 'input, 'output>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.