Struct ContractRequest 

pub struct ContractRequest<T: TensorSupported> {
    pub a: OperandSpec<T>,
    pub b: OperandSpec<T>,
    pub c: OperandSpec<T>,
    pub alpha: T,
    pub beta: T,
    pub compute: ComputeDesc,
    pub alignment: u32,
    pub reply: Sender<Result<(), GpuError>>,
}

Dtype-generic contraction request.

Fields

a: OperandSpec<T>

b: OperandSpec<T>

c: OperandSpec<T>

alpha: T

beta: T

compute: ComputeDesc

alignment: u32

Required tensor alignment in bytes.

reply: Sender<Result<(), GpuError>>

Implementations

impl<T: TensorSupported> ContractRequest<T>

pub fn new( a: OperandSpec<T>, b: OperandSpec<T>, c: OperandSpec<T>, alpha: T, beta: T, reply: Sender<Result<(), GpuError>>, ) -> Self

pub fn with_compute(self, compute: ComputeDesc) -> Self

Trait Implementations

impl<T: TensorSupported> TensorDispatch for ContractRequest<T>

fn op_tag(&self) -> &'static str

fn dtype_tag(&self) -> &'static str

fn dispatch(self: Box<Self>, ctx: &TensorDispatchCtx)

fn fail_mock(self: Box<Self>)