Searched defs:Kv (Results 1 – 3 of 3) sorted by relevance
351 int64_t Kv = value.size(3); in _efficient_attention_backward() local
1026 int64_t Kv = value.size(-1); in _efficient_attention_forward() local
5311 auto Kv = K.narrow_symint(-1, k, m - k).matmul(v); in householder_product_backward() local