Searched defs:head_dim_qk (Results 1 – 4 of 4) sorted by relevance
229 const int64_t head_dim_qk = query.size(3); in sdpa_nested_preprocessing_with_broadcast() local406 const int64_t head_dim_qk = query.size(3); in sdpa_nested_preprocessing() local483 const int64_t head_dim_qk = query.size(3); in sdpa_nested_preprocessing_backward() local
196 const int64_t head_dim_qk = query.size(3); in _scaled_dot_product_cudnn_attention_backward_cuda() local
754 const int64_t head_dim_qk = query.size(3); in _scaled_dot_product_cudnn_attention_cuda() local
456 const int64_t head_dim_qk = query.size(3); in custom_scaled_dot_product_fused_attention_overrideable() local