Searched defs:head_size (Results 1 – 8 of 8) sorted by relevance
298 const int num_heads, const int head_size, const int max_seqlen_k, const int max_seqlen_q, in set_params_splitkv()443 const int head_size = round_multiple(head_size_og, 8); in mha_fwd() local682 const int head_size = round_multiple(head_size_og, 8); in mha_varlen_fwd() local868 const int head_size = sizes[3]; in mha_bwd() local1090 const int head_size = sizes[2]; in mha_varlen_bwd() local1378 const int head_size = round_multiple(head_size_og, 8); in mha_fwd_kvcache() local
93 int64_t head_size; variable
172 const int head_size = 128; in y4m_header_parser() local
122 bool FontFileParser::analyzeFontRevision(const uint8_t* head_data, size_t head_size, in analyzeFontRevision()
62 size_t head_size = mask + 1 - from_pos; in CopyLiteralsToByteArray() local
63 size_t head_size = mask + 1 - from_pos; in CopyLiteralsToByteArray() local
899 pub const fn head_size(&self) -> DeviceSize { in head_size() method
637 size_t head_size = strlen(cmd_head); in ts_write() local