Skip to main content

KernelFactory

Trait KernelFactory 

Source
pub trait KernelFactory: Send + Sync {
    // Required method
    fn create(
        &self,
        node: &Node,
        input_shapes: &[Vec<usize>],
    ) -> Result<Box<dyn Kernel>>;
}
Expand description

Creates kernels for a specific op.

Required Methods§

Source

fn create( &self, node: &Node, input_shapes: &[Vec<usize>], ) -> Result<Box<dyn Kernel>>

Dyn Compatibility§

This trait is dyn compatible.

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

Implementors§