Searched refs:hash_table_shape (Results 1 – 1 of 1) sorted by relevance
221 std::vector<size_t> hash_table_shape = {cache_vocab_size, embedding_size}; in HashSwapOut() local224 …std::vector<std::vector<size_t>> input_shape = {hash_table_shape, swap_out_index_shape, offset_sha… in HashSwapOut()261 std::vector<size_t> hash_table_shape = {cache_vocab_size, embedding_size}; in HashSwapIn() local265 …std::vector<std::vector<size_t>> input_shape = {hash_table_shape, swap_in_index_shape, swap_in_val… in HashSwapIn()