Type Alias rust_bert::models::mbart::LayerState

pub type LayerState = LayerState;

Aliased Type

struct LayerState {
    pub prev_key: Tensor,
    pub prev_value: Tensor,
}

Fields

prev_key: Tensor

Cached keys

prev_value: Tensor

Cached values