Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dattention.cpp667 bool compute_logsumexp = should_compute_logsumexp(query_, key, value); in scaled_dot_product_attention() local
689 bool compute_logsumexp = should_compute_logsumexp(query_, key, value); in scaled_dot_product_attention() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dattention.cu968 bool compute_logsumexp, in _efficient_attention_forward()