Struct TensorMapInputs 

pub struct TensorMapInputs<Lhs: Numeric, Rhs: Numeric, EO: Numeric> {
    pub lhs: View<Line<Lhs>, Coords3d>,
    pub rhs: View<Line<Rhs>, Coords3d>,
    pub acc: CubeOption<View<Line<EO>, 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<EO>, Coords3d>>

The accumulator

Trait Implementations

impl<Lhs: Numeric, Rhs: Numeric, EO: Numeric> ConcreteInputsFactory for TensorMapInputs<Lhs, Rhs, EO>

fn create<'a, R: Runtime>( _client: &ComputeClient<R::Server>, lhs_handle: &'a MatmulInputHandleRef<'a, R>, rhs_handle: &'a MatmulInputHandleRef<'a, R>, selection: &MatmulSelection, problem: &MatmulProblem, line_sizes: &MatmulLineSizes, _config: impl BatchConfig, ) -> Self::RuntimeArg<'a, R>

impl<Lhs: Numeric, Rhs: Numeric, EO: Numeric> CubeType for TensorMapInputs<Lhs, Rhs, EO>

type ExpandType = TensorMapInputsExpand<Lhs, Rhs, EO>

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, EO: Numeric> LaunchArg for TensorMapInputs<Lhs, Rhs, EO>

type RuntimeArg<'a, R: Runtime> = TensorMapInputsLaunch<'a, Lhs, Rhs, EO, R>

The runtime argument for the kernel.

type CompilationArg = TensorMapInputsCompilationArg<Lhs, Rhs, EO>

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.