flashlight_tensor::wgpu

Enum GpuOperations

pub enum GpuOperations {
Show 34 variants    Add,
    TensAdd,
    Sub,
    TensSub,
    Mul,
    TensMul,
    Div,
    TensDiv,
    NLog,
    Log,
    Matmul,
    ReLU,
    Sigmoid,
    BroadcastAdd,
    BroadcastSub,
    BroadcastMul,
    BroadcastDiv,
    MatrixTranspose,
    MatrixRowSum,
    MatrixRowProd,
    MatrixColSum,
    MatrixColProd,
    ForwardWeightBiasNoActiv,
    ForwardWeightBiasSigmoid,
    ForwardWeightBiasRelu,
    BackpropWeightMergeNoActiv,
    BackpropWeightMergeSigmoid,
    BackpropWeightMergeRelu,
    BackpropBiasMergeNoActiv,
    BackpropBiasMergeRelu,
    BackpropBiasMergeSigmoid,
    BackpropGradientMergeNoActiv,
    BackpropGradientMergeRelu,
    BackpropGradientMergeSigmoid,
}

Expand description

Gpu tensor operations supported by this library

Variants§

Add

TensAdd

Sub

TensSub

Mul

TensMul

Div

TensDiv

NLog

Log

Matmul

ReLU

Sigmoid

BroadcastAdd

BroadcastSub

BroadcastMul

BroadcastDiv

MatrixTranspose

MatrixRowSum

MatrixRowProd

MatrixColSum

MatrixColProd

ForwardWeightBiasNoActiv

ForwardWeightBiasSigmoid

ForwardWeightBiasRelu

BackpropWeightMergeNoActiv

BackpropWeightMergeSigmoid

BackpropWeightMergeRelu

BackpropBiasMergeNoActiv

BackpropBiasMergeRelu

BackpropBiasMergeSigmoid

BackpropGradientMergeNoActiv

BackpropGradientMergeRelu

BackpropGradientMergeSigmoid

Trait Implementations§

impl Debug for GpuOperations

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

Formats the value using the given formatter. Read more

impl PartialEq for GpuOperations

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

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

1.0.0 · Source§

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 Eq for GpuOperations

impl StructuralPartialEq for GpuOperations

Auto Trait Implementations§

impl Freeze for GpuOperations

impl RefUnwindSafe for GpuOperations

impl Send for GpuOperations

impl Sync for GpuOperations

impl Unpin for GpuOperations

impl UnwindSafe for GpuOperations

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> Downcast<T> for T

fn downcast(&self) -> &T

impl<Q, K> Equivalent<K> for Q
where Q: Eq + ?Sized, K: Borrow<Q> + ?Sized,

fn equivalent(&self, key: &K) -> bool

Checks if this value is equivalent to the given key. Read more

impl<Q, K> Equivalent<K> for Q
where Q: Eq + ?Sized, K: Borrow<Q> + ?Sized,

fn equivalent(&self, key: &K) -> bool

Compare self to key and return true if they are equal.

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<T> Upcast<T> for T

fn upcast(&self) -> Option<&T>

impl<T> WasmNotSend for T
where T: Send,

impl<T> WasmNotSendSync for T
where T: WasmNotSend + WasmNotSync,

impl<T> WasmNotSync for T
where T: Sync,