Searched defs:compute_logsumexp (Results 1 – 2 of 2) sorted by relevance
667 bool compute_logsumexp = should_compute_logsumexp(query_, key, value); in scaled_dot_product_attention() local689 bool compute_logsumexp = should_compute_logsumexp(query_, key, value); in scaled_dot_product_attention() local
968 bool compute_logsumexp, in _efficient_attention_forward()