Struct RefBackend 

pub struct RefBackend;

Pure Rust reference implementation of KernelBackend.

Used for CI testing and platforms where no hardware-accelerated backend is available.

Trait Implementations

impl KernelBackend for RefBackend

fn conv2d(&mut self, params: Conv2dParams<'_>) -> Status

Execute a 2D convolution.

fn depthwise_conv2d(&mut self, params: DepthwiseConv2dParams<'_>) -> Status

Execute a depthwise 2D convolution.

fn fully_connected(&mut self, params: FullyConnectedParams<'_>) -> Status

Execute a fully-connected layer.

fn avg_pool(&mut self, params: PoolParams<'_>) -> Status

Execute average pooling.

fn max_pool(&mut self, params: PoolParams<'_>) -> Status

Execute max pooling.

fn softmax(&mut self, params: SoftmaxParams<'_>) -> Status

Execute softmax.

fn add(&mut self, params: ElementwiseAddParams<'_>) -> Status

Execute element-wise addition.

fn conv2d_scratch_size( input_shape: [usize; 4], weights_shape: [usize; 4], output_shape: [usize; 4], ) -> usize

where Self: Sized,

Returns the scratch buffer size in bytes required by Self::conv2d.

fn depthwise_conv2d_scratch_size( input_shape: [usize; 4], weights_shape: [usize; 4], output_shape: [usize; 4], ) -> usize

where Self: Sized,

Returns the scratch buffer size in bytes required by Self::depthwise_conv2d.

fn softmax_scratch_size(num_classes: usize) -> usize

where Self: Sized,

Returns the scratch buffer size in bytes required by Self::softmax.