Skip to main content

DflashLayer

memra_engine::dflash

Struct DflashLayer

pub struct DflashLayer {
    pub wq: GpuTensor,
    pub wk: GpuTensor,
    pub wv: GpuTensor,
    pub wo: GpuTensor,
    pub w_gate: GpuTensor,
    pub w_up: GpuTensor,
    pub w_down: GpuTensor,
    pub ln_in: CudaSlice<f32>,
    pub ln_post: CudaSlice<f32>,
    pub q_norm: CudaSlice<f32>,
    pub k_norm: CudaSlice<f32>,
}

Fields§

§wq: GpuTensor§wk: GpuTensor§wv: GpuTensor§wo: GpuTensor§w_gate: GpuTensor§w_up: GpuTensor§w_down: GpuTensor§ln_in: CudaSlice<f32>§ln_post: CudaSlice<f32>§q_norm: CudaSlice<f32>§k_norm: CudaSlice<f32>

Auto Trait Implementations§

impl Freeze for DflashLayer

impl RefUnwindSafe for DflashLayer

impl Send for DflashLayer

impl Sync for DflashLayer

impl Unpin for DflashLayer

impl UnsafeUnpin for DflashLayer

impl UnwindSafe for DflashLayer

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.