Searched refs:all_reduces (Results 1 – 2 of 2) sorted by relevance
66 std::vector<HloInstruction*> all_reduces; in MakeCrossReplicaReductions() local77 all_reduces.push_back(b->AddInstruction(HloInstruction::CreateAllReduce( in MakeCrossReplicaReductions()82 return b->AddInstruction(HloInstruction::CreateTuple(all_reduces)); in MakeCrossReplicaReductions()
97 all_reduces = []105 all_reduces.append(collective_ops.all_reduce(107 results = sess.run(all_reduces)