Struct ConvPlan 

pub struct ConvPlan { /* private fields */ }

Prepared MLAS Float32 convolution parameters for one concrete NCHW shape.

Implementations

impl ConvPlan

pub fn new( batch_count: usize, group_count: usize, input_channels_per_group: usize, input_shape: &[i64], kernel_shape: &[i64], dilation_shape: &[i64], padding: &[i64], stride_shape: &[i64], output_shape: &[i64], filter_count_per_group: usize, ) -> Option<Self>

Prepare an N-dimensional NCHW convolution and return its scratch size.

pub fn working_buffer_elements(&self) -> usize

Number of Float32 scratch elements required by Self::run.

pub fn run( &self, input: &[f32], filter: &[f32], bias: Option<&[f32]>, working_buffer: &mut [f32], output: &mut [f32], )

Execute the prepared convolution.

Trait Implementations

impl Drop for ConvPlan

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.

impl Send for ConvPlan

impl Sync for ConvPlan