Searched defs:head_size_rounded (Results 1 – 1 of 1) sorted by relevance
299 const int head_size_rounded, const float p_dropout, in set_params_splitkv()444 const int head_size_rounded = round_multiple(head_size, 32); in mha_fwd() local683 const int head_size_rounded = round_multiple(head_size, 32); in mha_varlen_fwd() local881 const int head_size_rounded = round_multiple(head_size, 32); in mha_bwd() local1103 const int head_size_rounded = round_multiple(head_size, 32); in mha_varlen_bwd() local1379 const int head_size_rounded = round_multiple(head_size, 32); in mha_fwd_kvcache() local