Searched defs:compute_log_sumexp (Results 1 – 5 of 5) sorted by relevance
543 bool compute_log_sumexp, in _scaled_dot_product_efficient_attention_batch_rule()584 bool compute_log_sumexp, in _scaled_dot_product_cudnn_attention_batch_rule()
287 bool compute_log_sumexp, in _scaled_dot_product_efficient_attention_nestedtensor_cuda()
605 int compute_log_sumexp, in aoti_torch__scaled_dot_product_efficient_attention()
794 bool compute_log_sumexp, in _scaled_dot_product_efficient_attention_cuda()
9655 def fn(q, k, v, attn_bias, compute_log_sumexp): argument