Searched defs:s1_split_num (Results 1 – 2 of 2) sorted by relevance
50 int64_t s1_split_num() { return s1_split_num_; } in s1_split_num() function
296 int64_t s1_split_num = basic_stra[s_split_idx]; in PrepareFlashAttentionScore() local