Struct rust_bert::models::t5::LayerState
source · pub struct LayerState {
pub prev_key: Tensor,
pub prev_value: Tensor,
}
Expand description
Cache for T5 attention layers
Stores the cached value of key, value and key to avoid recalculation (e.g. at each generation step)
Fields§
§prev_key: Tensor
Cached keys
prev_value: Tensor
Cached values
Trait Implementations§
source§impl Clone for LayerState
impl Clone for LayerState
Auto Trait Implementations§
impl RefUnwindSafe for LayerState
impl Send for LayerState
impl !Sync for LayerState
impl Unpin for LayerState
impl UnwindSafe for LayerState
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