Skip to main content

ConcreteOutputFactory

Trait ConcreteOutputFactory 

Source
pub trait ConcreteOutputFactory: LaunchArg {
    // Required method
    fn create<'a, R: Runtime>(
        client: &ComputeClient<R::Server>,
        out: &'a TensorHandleRef<'a, R>,
        selection: &MatmulSelection,
        problem: &ConvolutionProblem,
        line_sizes: &MatmulLineSizes,
        config: impl ConvGemmConfig,
    ) -> Self::RuntimeArg<'a, R>;
}
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<'a, R: Runtime>( client: &ComputeClient<R::Server>, out: &'a TensorHandleRef<'a, R>, selection: &MatmulSelection, problem: &ConvolutionProblem, line_sizes: &MatmulLineSizes, config: impl ConvGemmConfig, ) -> Self::RuntimeArg<'a, R>

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: Numeric> ConcreteOutputFactory for View<Line<EG>, Coords3d, ReadWrite>

Source§

fn create<'a, R: Runtime>( client: &ComputeClient<R::Server>, out: &'a TensorHandleRef<'a, R>, _selection: &MatmulSelection, problem: &ConvolutionProblem, line_sizes: &MatmulLineSizes, config: impl ConvGemmConfig, ) -> Self::RuntimeArg<'a, R>

Implementors§