Struct ConvolutionConfig 

pub struct ConvolutionConfig {
    pub conv_type: ConvolutionType,
    pub input_dims: Vec<usize>,
    pub output_dims: Vec<usize>,
    pub kernel_dims: Vec<usize>,
    pub strides: Vec<usize>,
    pub padding: Vec<usize>,
    pub dilation: Vec<usize>,
    pub groups: usize,
    pub padding_mode: PaddingMode,
    pub dtype: DType,
    pub algorithm: ConvolutionAlgorithm,
}

Convolution configuration

Fields

conv_type: ConvolutionType

Convolution type

input_dims: Vec<usize>

Input dimensions [batch, channels, height, width] for 2D

output_dims: Vec<usize>

Output dimensions [batch, channels, height, width] for 2D

kernel_dims: Vec<usize>

Kernel dimensions [out_channels, in_channels, height, width] for 2D

strides: Vec<usize>

Stride in each dimension

padding: Vec<usize>

Padding in each dimension

dilation: Vec<usize>

Dilation in each dimension

groups: usize

Number of groups for grouped convolution

padding_mode: PaddingMode

Padding mode

dtype: DType

Data type

algorithm: ConvolutionAlgorithm

Preferred algorithm

Implementations

impl ConvolutionConfig

pub fn conv2d( batch_size: usize, in_channels: usize, out_channels: usize, input_size: (usize, usize), kernel_size: (usize, usize), stride: (usize, usize), padding: (usize, usize), ) -> Self

Create a new 2D convolution configuration

pub fn depthwise_conv2d( batch_size: usize, channels: usize, input_size: (usize, usize), kernel_size: (usize, usize), stride: (usize, usize), padding: (usize, usize), ) -> Self

Create a depthwise convolution configuration

pub fn with_algorithm(self, algorithm: ConvolutionAlgorithm) -> Self

Set the algorithm preference

pub fn with_dtype(self, dtype: DType) -> Self

Set the data type

pub fn with_dilation(self, dilation: Vec<usize>) -> Self

Set dilation

pub fn input_elements(&self) -> usize

Calculate total input elements

pub fn output_elements(&self) -> usize

Calculate total output elements

pub fn kernel_elements(&self) -> usize

Calculate total kernel elements

pub fn input_buffer_size(&self) -> usize

Get input buffer size in bytes

pub fn output_buffer_size(&self) -> usize

Get output buffer size in bytes

pub fn kernel_buffer_size(&self) -> usize

Get kernel buffer size in bytes

pub fn is_valid(&self) -> bool

Check if the configuration is valid

Trait Implementations

impl Clone for ConvolutionConfig

fn clone(&self) -> ConvolutionConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ConvolutionConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.