↓ 6 callersMethod_set_input_buffer(
self,
incremental_state: Dict[str, Dict[str, Optional[Tensor]]],
buffer: Dict[str, O
fairseq/fairseq/model_parallel/modules/multihead_attention.py:344
↓ 6 callersMethodinference_step(
self, generator, models, sample, prefix_tokens=None, constraints=None
)
fairseq/fairseq/tasks/fairseq_task.py:523
↓ 5 callersMethod_set_input_buffer(
self,
incremental_state: Dict[str, Dict[str, Optional[Tensor]]],
buffer: Dict[str, O
fairseq/examples/linformer/linformer_src/modules/multihead_linear_attention.py:442