Searched refs:shard_ids (Results 1 – 2 of 2) sorted by relevance
86 std::vector<uint32> shard_ids(ag->replica_groups().size() * in DecomposeAllGather() local90 shard_ids[group.replica_ids(i)] = i; in DecomposeAllGather()94 LiteralUtil::CreateR1<uint32>(shard_ids))); in DecomposeAllGather()
589 std::unordered_map<string, int32> shard_ids; member684 auto result = merge_state->shard_ids.insert( in MergeOneBundle()686 merge_state->shard_ids.size()}); in MergeOneBundle()705 for (const auto& p : merge.shard_ids) { in MergeBundles()707 << DataFilename(merged_prefix, p.second, merge.shard_ids.size()); in MergeBundles()710 DataFilename(merged_prefix, p.second, merge.shard_ids.size()))); in MergeBundles()