Trait ConcreteInputsFactory 

pub trait ConcreteInputsFactory<A: BatchMatmulRoutine<RuntimeArgs>>: LaunchArg {
    // Required method
    fn create<R: Runtime>(
        out_grad: InputBinding<R>,
        weights: InputBinding<R>,
        blueprint: &A::Blueprint,
        problem: &ConvolutionProblem,
        dtypes: &MatmulElems,
    ) -> (Self::RuntimeArg<R>, RuntimeArgsLaunch<R>);
}

Create the input runtime arguments for a matmul kernel that works on concrete inputs and output (not fused).

Required Methods

fn create<R: Runtime>( out_grad: InputBinding<R>, weights: InputBinding<R>, blueprint: &A::Blueprint, problem: &ConvolutionProblem, dtypes: &MatmulElems, ) -> (Self::RuntimeArg<R>, RuntimeArgsLaunch<R>)

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementations on Foreign Types

impl<Lhs: CubePrimitive, Rhs: CubePrimitive, EO: CubePrimitive, A: BatchMatmulRoutine<RuntimeArgs, Blueprint = BatchMatmulBlueprint>> ConcreteInputsFactory<A> for TensorMapInputs<Lhs, Rhs, EO>

fn create<R: Runtime>( out_grad: InputBinding<R>, weights: InputBinding<R>, blueprint: &BatchMatmulBlueprint, problem: &ConvolutionProblem, dtypes: &MatmulElems, ) -> (Self::RuntimeArg<R>, RuntimeArgsLaunch<R>)

impl<Lhs: CubePrimitive, Rhs: CubePrimitive, EO: CubePrimitive, A: BatchMatmulRoutine<RuntimeArgs>> ConcreteInputsFactory<A> for TensorInputs<Lhs, Rhs, EO>

fn create<R: Runtime>( out_grad: InputBinding<R>, weights: InputBinding<R>, blueprint: &A::Blueprint, problem: &ConvolutionProblem, _dtypes: &MatmulElems, ) -> (Self::RuntimeArg<R>, RuntimeArgsLaunch<R>)

Implementors