Searched defs:cumulative_sequence_length_q (Results 1 – 4 of 4) sorted by relevance
251 Tensor cumulative_sequence_length_q; in sdpa_nested_preprocessing_with_broadcast() local464 const Tensor& cumulative_sequence_length_q, in sdpa_nested_preprocessing_backward()
333 const Tensor& cumulative_sequence_length_q, in _scaled_dot_product_flash_attention_backward_nested()
68 const Tensor& cumulative_sequence_length_q, in _flash_attention_backward()739 const Tensor& cumulative_sequence_length_q, in _scaled_dot_product_flash_attention_backward_cuda()
847 const std::optional<Tensor>& cumulative_sequence_length_q, in _flash_attention_forward()