Struct TensorInputs

pub struct TensorInputs<EG: Numeric> {
    pub lhs: Tensor<Line<EG>>,
    pub rhs: Tensor<Line<EG>>,
}

Input representation for TensorArgs implementing MatmulArgs.

Fields

lhs: Tensor<Line<EG>>

The lhs tensor.

rhs: Tensor<Line<EG>>

The rhs tensor.

Trait Implementations

impl<EG: Numeric> ConcreteInputsFactory for TensorInputs<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 TensorInputs<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 TensorInputs<EG>

type ExpandType = TensorInputsExpand<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 TensorInputs<EG>

type RuntimeArg<'a, R: Runtime> = TensorInputsLaunch<'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 TensorInputs<EG>

type CompilationArg = TensorInputsCompilationArg<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.