Searched refs:sdpa_nested_preprocessing (Results 1 – 3 of 3) sorted by relevance
13 sdpa_nested_preprocessing(
245 output_shape] = preprocessing::sdpa_nested_preprocessing(query, key, value); in _scaled_dot_product_flash_attention_nestedtensor_cuda()299 output_shape] = preprocessing::sdpa_nested_preprocessing(query, key, value); in _scaled_dot_product_efficient_attention_nestedtensor_cuda()
385 sdpa_nested_preprocessing( in sdpa_nested_preprocessing() function