Struct DepthwiseCost 

pub struct DepthwiseCost {
    pub batch: usize,
    pub channels: usize,
    pub size: usize,
    pub out_size: usize,
    pub kernel: usize,
    pub dtype: ElemType,
}

Minimal representation of a depthwise convolution’s cost. Each channel has its own filter and contracts over nothing else, so this is not the implicit gemm Conv2dCost describes.

Fields

batch: usize

channels: usize

size: usize

The input’s spatial extent, square.

out_size: usize

The output’s spatial extent, square.

kernel: usize

The filter’s spatial extent, square.

dtype: ElemType

Element type of the maps and the filter.

Implementations

impl DepthwiseCost

pub fn work(&self) -> Work

Calculates the compute operations and compulsory memory traffic for the convolution.

pub fn compute_ops(&self) -> usize

2 * taps - 1 per output element: one multiply per tap and the adds that join them.

pub fn traffic(&self) -> (usize, usize)

Compulsory global traffic in bytes, split by direction, which work sums.

pub fn compute_key(&self) -> ThroughputKey

A depthwise pass contracts over one filter window and issues no MMA, so its arithmetic is the scalar peak’s.

Trait Implementations

impl Clone for DepthwiseCost

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for DepthwiseCost

impl Debug for DepthwiseCost

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

Formats the value using the given formatter.