Searched refs:swap_in_value_shape (Results 1 – 1 of 1) sorted by relevance
263 std::vector<size_t> swap_in_value_shape = {swap_in_size, embedding_size}; in HashSwapIn() local265 …tor<std::vector<size_t>> input_shape = {hash_table_shape, swap_in_index_shape, swap_in_value_shape, in HashSwapIn()