def custom_sdpa_with_start_pos_forward()

in optimum/executorch/attentions/custom_sdpa.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

def custom_sdpa_with_start_pos_forward(
    module: torch.nn.Module,
    query: torch.Tensor,
    key: torch.Tensor,
    value: torch.Tensor,
    attention_mask: Union[torch.Tensor, "BlockMask"],  # noqa
    scaling: Optional[float] = None,
    softcap: Optional[float] = None,
    head_mask: Optional[torch.Tensor] = None,
    **kwargs,