Struct dfdx::nn::builders::TransformerEncoderBlock
source · pub struct TransformerEncoderBlock<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize>;
Trait Implementations§
source§impl<const M: usize, const H: usize, const F: usize, E: Dtype, D: Device<E>> BuildOnDevice<D, E> for TransformerEncoderBlock<M, H, F>where
TransformerEncoderBlock<M, H, F, E, D>: BuildModule<D, E>,
impl<const M: usize, const H: usize, const F: usize, E: Dtype, D: Device<E>> BuildOnDevice<D, E> for TransformerEncoderBlock<M, H, F>where TransformerEncoderBlock<M, H, F, E, D>: BuildModule<D, E>,
type Built = TransformerEncoderBlock<M, H, F, E, D>
fn try_build_on_device(device: &D) -> Result<Self::Built, D::Err>
fn build_on_device(device: &D) -> Self::Built
Auto Trait Implementations§
impl<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize> RefUnwindSafe for TransformerEncoderBlock<MODEL_DIM, NUM_HEADS, FF_DIM>
impl<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize> Send for TransformerEncoderBlock<MODEL_DIM, NUM_HEADS, FF_DIM>
impl<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize> Sync for TransformerEncoderBlock<MODEL_DIM, NUM_HEADS, FF_DIM>
impl<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize> Unpin for TransformerEncoderBlock<MODEL_DIM, NUM_HEADS, FF_DIM>
impl<const MODEL_DIM: usize, const NUM_HEADS: usize, const FF_DIM: usize> UnwindSafe for TransformerEncoderBlock<MODEL_DIM, NUM_HEADS, FF_DIM>
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more