Enum KernelType 

pub enum KernelType {
    SGDUpdate,
    AdamUpdate,
    AdamWUpdate,
    RMSpropUpdate,
    AdaGradUpdate,
    MomentumUpdate,
    GradientClipping,
    VectorOps,
    MatrixOps,
    Reduction,
}

Kernel types for different optimization algorithms

Variants

SGDUpdate

SGD update kernel

AdamUpdate

Adam update kernel

AdamWUpdate

AdamW update kernel

RMSpropUpdate

RMSprop update kernel

AdaGradUpdate

AdaGrad update kernel

MomentumUpdate

Momentum update kernel

GradientClipping

Gradient clipping kernel

VectorOps

Vector operations

MatrixOps

Matrix operations

Reduction

Reduction operations

Trait Implementations

impl Clone for KernelType

fn clone(&self) -> KernelType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for KernelType

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

Formats the value using the given formatter.

impl Hash for KernelType

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for KernelType

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

impl Eq for KernelType

impl StructuralPartialEq for KernelType