Type Definition rust_bert::pegasus::LayerState

pub type LayerState = BartLayerState;

Cache for Pegasus attention layers

Stores the cached value of key, value and key padding mask to avoid recalculation (e.g. at each generation step) Identical to BART cache (type alias).