Searched defs:query_buffer_reshaped (Results 1 – 1 of 1) sorted by relevance
293 Tensor query_buffer_reshaped; in sdpa_nested_preprocessing_with_broadcast() local416 Tensor query_buffer_reshaped; in sdpa_nested_preprocessing() local495 Tensor query_buffer_reshaped; in sdpa_nested_preprocessing_backward() local