Searched defs:cache_batch_idx (Results 1 – 2 of 2) sorted by relevance
101 int * __restrict__ cache_batch_idx; member
1482 auto cache_batch_idx = cache_batch_idx_.value(); in mha_fwd_kvcache() local