Struct TensorMapInputs

pub struct TensorMapInputs<EG: Numeric> {
    pub lhs: TensorMap<EG>,
    pub rhs: TensorMap<EG>,
}

Input representation for TensorArgs implementing MatmulArgs.

Fields

lhs: TensorMap<EG>

The lhs tensor.

rhs: TensorMap<EG>

The rhs tensor.

Trait Implementations

impl<EG: Numeric> ConcreteInputsFactory for TensorMapInputs<EG>

fn create<'a, R: Runtime>( lhs: &'a TensorHandleRef<'a, R>, rhs: &'a TensorHandleRef<'a, R>, selection: &MatmulSelection, problem: &MatmulProblem, ) -> Self::RuntimeArg<'a, R>

impl<EI: Numeric> ConvInputsLaunch for TensorMapInputs<EI>

fn create<'a, R: Runtime>( lhs: &'a TensorHandleRef<'a, R>, rhs: &'a TensorHandleRef<'a, R>, selection: &MatmulSelection, problem: &ConvolutionProblem, ) -> Self::RuntimeArg<'a, R>

impl<EG: Numeric> CubeType for TensorMapInputs<EG>

type ExpandType = TensorMapInputsExpand<EG>

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

Wrapper around the init method, necessary to type inference.

impl<EG: Numeric> LaunchArg for TensorMapInputs<EG>

type RuntimeArg<'a, R: Runtime> = TensorMapInputsLaunch<'a, EG, R>

The runtime argument for the kernel.

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

impl<EG: Numeric> LaunchArgExpand for TensorMapInputs<EG>

type CompilationArg = TensorMapInputsCompilationArg<EG>

Compilation argument.

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.