Struct DefaultConvolutionOps 

pub struct DefaultConvolutionOps { /* private fields */ }

Default convolution operations implementation

Implementations

impl DefaultConvolutionOps

pub fn new() -> Self

pub fn with_performance_hints(self, hints: ConvolutionPerformanceHints) -> Self

Trait Implementations

impl ConvolutionOps for DefaultConvolutionOps

fn convolution<'life0, 'life1, 'life2, 'life3, 'life4, 'life5, 'life6, 'async_trait>( &'life0 self, _device: &'life1 Device, _input: &'life2 Buffer, _kernel: &'life3 Buffer, _bias: Option<&'life4 Buffer>, _output: &'life5 Buffer, _config: &'life6 ConvolutionConfig, ) -> Pin<Box<dyn Future<Output = BackendResult<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, 'life4: 'async_trait, 'life5: 'async_trait, 'life6: 'async_trait,

Execute a convolution operation

fn conv2d<'life0, 'life1, 'life2, 'life3, 'life4, 'life5, 'async_trait>( &'life0 self, _device: &'life1 Device, _input: &'life2 Buffer, _kernel: &'life3 Buffer, _bias: Option<&'life4 Buffer>, _output: &'life5 Buffer, _stride: (usize, usize), _padding: (usize, usize), _dilation: (usize, usize), ) -> Pin<Box<dyn Future<Output = BackendResult<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, 'life4: 'async_trait, 'life5: 'async_trait,

Execute a 2D convolution

fn depthwise_conv2d<'life0, 'life1, 'life2, 'life3, 'life4, 'life5, 'async_trait>( &'life0 self, _device: &'life1 Device, _input: &'life2 Buffer, _kernel: &'life3 Buffer, _bias: Option<&'life4 Buffer>, _output: &'life5 Buffer, _stride: (usize, usize), _padding: (usize, usize), ) -> Pin<Box<dyn Future<Output = BackendResult<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, 'life4: 'async_trait, 'life5: 'async_trait,

Execute a depthwise convolution

fn conv_transpose2d<'life0, 'life1, 'life2, 'life3, 'life4, 'life5, 'async_trait>( &'life0 self, _device: &'life1 Device, _input: &'life2 Buffer, _kernel: &'life3 Buffer, _bias: Option<&'life4 Buffer>, _output: &'life5 Buffer, _stride: (usize, usize), _padding: (usize, usize), _output_padding: (usize, usize), ) -> Pin<Box<dyn Future<Output = BackendResult<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, 'life4: 'async_trait, 'life5: 'async_trait,

Execute a transposed convolution

fn grouped_conv2d<'life0, 'life1, 'life2, 'life3, 'life4, 'life5, 'async_trait>( &'life0 self, _device: &'life1 Device, _input: &'life2 Buffer, _kernel: &'life3 Buffer, _bias: Option<&'life4 Buffer>, _output: &'life5 Buffer, _groups: usize, _stride: (usize, usize), _padding: (usize, usize), ) -> Pin<Box<dyn Future<Output = BackendResult<()>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, 'life4: 'async_trait, 'life5: 'async_trait,

Execute a grouped convolution

fn select_algorithm(&self, config: &ConvolutionConfig) -> ConvolutionAlgorithm

Get the best algorithm for given configuration

fn supports_convolution(&self) -> bool

Check if convolution operations are supported

fn supported_conv_types(&self) -> Vec<ConvolutionType>

Get supported convolution types

fn supported_algorithms(&self) -> Vec<ConvolutionAlgorithm>

Get supported algorithms

impl Default for DefaultConvolutionOps

fn default() -> Self

Returns the “default value” for a type.