Enum OpType 

pub enum OpType {
    Identity,
    Conv3x3,
    Conv5x5,
    DilatedConv3x3,
    DepthwiseSep3x3,
    MaxPool3x3,
    AvgPool3x3,
    Skip,
    Zero,
    Linear(usize),
    GRU,
    LSTM,
}

Operation types in a NAS cell

Variants

Identity

Pass-through with no transformation

Conv3x3

3x3 convolution

Conv5x5

5x5 convolution

DilatedConv3x3

Dilated 3x3 convolution (dilation=2)

DepthwiseSep3x3

Depthwise separable 3x3 convolution

MaxPool3x3

3x3 max pooling

AvgPool3x3

3x3 average pooling

Skip

Skip connection (residual)

Zero

Zero connection (no gradient flow)

Linear(usize)

Fully-connected layer with given output size

GRU

Gated Recurrent Unit cell

LSTM

Long Short-Term Memory cell

Implementations

impl OpType

pub fn num_params(&self, in_channels: usize) -> usize

Approximate number of trainable parameters for this op.

pub fn flops(&self, in_channels: usize, spatial: usize) -> usize

Approximate FLOPs for this op given input channels and spatial size.

Trait Implementations

impl Clone for OpType

fn clone(&self) -> OpType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OpType

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

Formats the value using the given formatter.

impl PartialEq for OpType

fn eq(&self, other: &OpType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for OpType