Skip to main content

ConcreteOutputFactory

Trait ConcreteOutputFactory 

Source
pub trait ConcreteOutputFactory<A: BatchMatmulRoutine<RuntimeArgs>>: LaunchArg {
    // Required method
    fn create(
        out: TensorBinding,
        blueprint: &A::Blueprint,
        problem: &ConvolutionProblem,
    ) -> Self::RuntimeArg;
}
Expand description

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

Required Methods§

Source

fn create( out: TensorBinding, blueprint: &A::Blueprint, problem: &ConvolutionProblem, ) -> Self::RuntimeArg

Dyn Compatibility§

This trait is not dyn compatible.

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

Implementations on Foreign Types§

Source§

impl<EG: CubePrimitive, A: BatchMatmulRoutine<RuntimeArgs>> ConcreteOutputFactory<A> for TensorOutput<EG>

Source§

fn create( out: TensorBinding, blueprint: &A::Blueprint, problem: &ConvolutionProblem, ) -> Self::RuntimeArg

Implementors§