Enum coaster_nn::ConvBackwardFilterAlgo

pub enum ConvBackwardFilterAlgo {
    Auto,
    ImplicitGEMM,
    ImplicitGEMMSum,
    ImplicitPrecompiledGEMMSum,
    FFT,
    WinogradNonFused,
}

Different algorithms to compute the gradient with respect to the filter.

Variants

Auto

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

ImplicitGEMM

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

Does not need any memory workspace.

The results are deterministic.

ImplicitGEMMSum

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

Does not need any memory workspace.

The results are non-deterministic.

ImplicitPrecompiledGEMMSum

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

The results are non-deterministic.

FFT

Compute the convolution as Fast-Fourier Transform.

Needs a significant memory workspace.

The results are deterministic.

WinogradNonFused

Winograd Transform Non-Fused

Implementations

impl ConvBackwardFilterAlgo

pub fn is_auto(&self) -> bool

Check if algorithim should be chosen automatically.

Trait Implementations

impl Clone for ConvBackwardFilterAlgo

fn clone(&self) -> ConvBackwardFilterAlgo

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ConvBackwardFilterAlgo

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

Formats the value using the given formatter.

impl PartialEq for ConvBackwardFilterAlgo

fn eq(&self, other: &ConvBackwardFilterAlgo) -> 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 ConvBackwardFilterAlgo

impl Eq for ConvBackwardFilterAlgo

impl StructuralPartialEq for ConvBackwardFilterAlgo