Trait ConcreteInputsFactory 

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

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

Required Methods

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

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( out_grad: InputBinding, weights: InputBinding, blueprint: &BatchMatmulBlueprint, problem: &ConvolutionProblem, dtypes: &MatmulElems, ) -> (Self::RuntimeArg, RuntimeArgsLaunch)

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

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

Implementors