Searched refs:MakeCrossReplicaReductions (Results 1 – 1 of 1) sorted by relevance
62 HloInstruction* MakeCrossReplicaReductions( in MakeCrossReplicaReductions() function120 auto root = MakeCrossReplicaReductions( in TEST_F()167 MakeCrossReplicaReductions( in TEST_F()188 MakeCrossReplicaReductions({8, 4}, {sum, sum}, &inputs, &b); in TEST_F()