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>>,
    pub acc_batch: CubeOption<VirtualLayout<Coords1d, Coords1d>>,
}

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

acc_batch: CubeOption<VirtualLayout<Coords1d, Coords1d>>

The accumulator batch layout

Trait Implementations

impl<Lhs: Clone + Numeric, Rhs: Clone + Numeric, EO: Clone + Numeric> Clone for TensorMapInputs<Lhs, Rhs, EO>

fn clone(&self) -> TensorMapInputs<Lhs, Rhs, EO>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

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, dtypes: &MatmulElems, ) -> 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.

impl<Lhs: Copy + Numeric, Rhs: Copy + Numeric, EO: Copy + Numeric> Copy for TensorMapInputs<Lhs, Rhs, EO>