Enum coaster_nn::ConvForwardAlgo

pub enum ConvForwardAlgo {
    Auto,
    GEMM,
    ImplicitGEMM,
    ImplicitPrecompiledGEMM,
    FFT,
    FFTTiling,
    Direct,
    Winograd,
    WinogradNonFused,
}

Different algorithms to compute the convolution forward algorithm.

Variants

Auto

Attempt to automatically find the best algorithm of all the other available ones.

GEMM

Compute the convolution as explicit matrix product.

Needs a significant memory workspace.

ImplicitGEMM

Compute the convolution as matrix product without forming the matrix that holds the input data.

Does not need any memory workspace.

ImplicitPrecompiledGEMM

Similar to ImplicitGEMM but needs some workspace to precompile the implicit indices.

FFT

Compute the convolution as Fast-Fourier Transform.

Needs a significant memory workspace.

FFTTiling

Compute the convolution as Fast-Fourier Transform with 32x32 tiles.

Needs a significant memory workspace.

Direct

Compute the convolution without implicit or explicit matrix-multiplication. Do not try to use this.

Listed in cuDNN docs but cuDNN does not provide a implementation.

Winograd

Winograd Transform

WinogradNonFused

Winograd Transform Non-Fused

Implementations

impl ConvForwardAlgo

pub fn is_auto(&self) -> bool

Check if algorithim should be chosen automatically.

Trait Implementations

impl Clone for ConvForwardAlgo

fn clone(&self) -> ConvForwardAlgo

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ConvForwardAlgo

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

Formats the value using the given formatter.

impl PartialEq for ConvForwardAlgo

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

This method tests for self and other values to be equal, and is used by ==.

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

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

impl Copy for ConvForwardAlgo

impl Eq for ConvForwardAlgo

impl StructuralPartialEq for ConvForwardAlgo