Home
last modified time | relevance | path

Searched defs:alibi_slopes (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/
Dattention_backward.cu96 std::optional<at::Tensor> alibi_slopes{std::nullopt}; in _flash_attention_backward() local
Dattention.cu866 std::optional<Tensor> alibi_slopes = _alibi_slopes; in _flash_attention_forward() local