Searched defs:kcache (Results 1 – 5 of 5) sorted by relevance
998 struct r600_bytecode_kcache *kcache, in r600_bytecode_alloc_kcache_line()1060 struct r600_bytecode_kcache *kcache, in r600_bytecode_alloc_inst_kcache_lines()1084 struct r600_bytecode_kcache * kcache) in r600_bytecode_assign_kcache_banks()1124 struct r600_bytecode_kcache *kcache = kcache_sets; in r600_bytecode_alloc_kcache_lines() local
191 struct r600_bytecode_kcache kcache[4]; member
359 auto kcache = m_kcache; in try_reserve_kcache() local379 auto kcache = m_kcache; in try_reserve_kcache() local
207 ValueFactory::uniform(uint32_t index, int chan, int kcache) in uniform()
1254 …const at::Tensor &kcache, // batch_size_c x seqlen_k x num_heads_k x head_size or num_b… in mha_fwd_kvcache()