Searched refs:sizes_in_kib (Results 1 – 1 of 1) sorted by relevance
63 std::vector<int64> sizes_in_kib, std::vector<HloComputation*> reductions, in MakeCrossReplicaReductions() argument65 CHECK_EQ(reductions.size(), sizes_in_kib.size()); in MakeCrossReplicaReductions()67 for (int i = 0; i < sizes_in_kib.size(); i++) { in MakeCrossReplicaReductions()68 int64 size_in_kib = sizes_in_kib[i]; in MakeCrossReplicaReductions()