Struct TensorInputs 

pub struct TensorInputs<Lhs: Numeric, Rhs: Numeric, Acc: Numeric> {
    pub lhs: View<Line<Lhs>, Coords3d>,
    pub rhs: View<Line<Rhs>, Coords3d>,
    pub acc: CubeOption<View<Line<Acc>, Coords3d>>,
}

Input representation for TensorArgs implementing MatmulArgs.

Fields

lhs: View<Line<Lhs>, Coords3d>

The lhs tensor.

rhs: View<Line<Rhs>, Coords3d>

The rhs tensor.

acc: CubeOption<View<Line<Acc>, Coords3d>>

The tensor for loading the accumulator, if present

Trait Implementations

impl<Lhs: Numeric, Rhs: Numeric, Acc: Numeric> ConcreteInputsFactory for TensorInputs<Lhs, Rhs, Acc>

fn create<'a, R: Runtime>( client: &ComputeClient<R::Server>, lhs: &'a MatmulInputHandleRef<'a, R>, rhs: &'a MatmulInputHandleRef<'a, R>, _selection: &MatmulSelection, problem: &MatmulProblem, line_sizes: &MatmulLineSizes, config: impl BatchConfig, ) -> Self::RuntimeArg<'a, R>

impl<Lhs: Numeric, Rhs: Numeric, Acc: Numeric> CubeType for TensorInputs<Lhs, Rhs, Acc>

type ExpandType = TensorInputsExpand<Lhs, Rhs, Acc>

fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl<Lhs: Numeric, Rhs: Numeric, Acc: Numeric> LaunchArg for TensorInputs<Lhs, Rhs, Acc>

type RuntimeArg<'a, R: Runtime> = TensorInputsLaunch<'a, Lhs, Rhs, Acc, R>

The runtime argument for the kernel.

type CompilationArg = TensorInputsCompilationArg<Lhs, Rhs, Acc>

Compilation argument.

fn compilation_arg<'a, R: Runtime>( runtime_arg: &Self::RuntimeArg<'a, R>, ) -> Self::CompilationArg

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an input variable during compilation that fill the KernelBuilder.

fn expand_output( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register an output variable during compilation that fill the KernelBuilder.