Struct coaster_nn::frameworks::cuda::RnnSequenceDescriptors

pub struct RnnSequenceDescriptors {
    pub x_desc: Vec<TensorDescriptor>,
    pub y_desc: Vec<TensorDescriptor>,
    pub dx_desc: Vec<TensorDescriptor>,
    pub dy_desc: Vec<TensorDescriptor>,
    pub hx_desc: TensorDescriptor,
    pub cx_desc: TensorDescriptor,
    pub hy_desc: TensorDescriptor,
    pub cy_desc: TensorDescriptor,
    pub dhx_desc: TensorDescriptor,
    pub dcx_desc: TensorDescriptor,
    pub dhy_desc: TensorDescriptor,
    pub dcy_desc: TensorDescriptor,
}

Expand description

Struct to hold all Sequence Descriptors for an RNN Pass

Fields§

§x_desc: Vec<TensorDescriptor>

Input Descriptor

§y_desc: Vec<TensorDescriptor>

Output Descriptor

§dx_desc: Vec<TensorDescriptor>

Gradient Input Descriptor

§dy_desc: Vec<TensorDescriptor>

Gradient Output Descriptor

§hx_desc: TensorDescriptor

Hidden Input Descriptor

§cx_desc: TensorDescriptor

Cell Input Descriptor

§hy_desc: TensorDescriptor

Hidden Output Descriptor

§cy_desc: TensorDescriptor

Cell Output Descriptor

§dhx_desc: TensorDescriptor

Gradient Hidden Input Descriptor

§dcx_desc: TensorDescriptor

Gradient Cell Input Descriptor

§dhy_desc: TensorDescriptor

Gradient Hidden Output Descriptor

§dcy_desc: TensorDescriptor

Gradient Cell Output Descriptor

Trait Implementations§

impl Debug for RnnSequenceDescriptors

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

Formats the value using the given formatter. Read more

Auto Trait Implementations§

impl RefUnwindSafe for RnnSequenceDescriptors

impl !Send for RnnSequenceDescriptors

impl !Sync for RnnSequenceDescriptors

impl Unpin for RnnSequenceDescriptors

impl UnwindSafe for RnnSequenceDescriptors

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> 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<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V