Searched defs:head_size_og (Results 1 – 1 of 1) sorted by relevance
391 const int head_size_og = sizes[3]; in mha_fwd() local605 const int head_size_og = sizes[2]; in mha_varlen_fwd() local867 const int head_size_og = dout.size(3); in mha_bwd() local1089 const int head_size_og = dout.size(2); in mha_varlen_bwd() local1311 const int head_size_og = sizes[3]; in mha_fwd_kvcache() local