Searched refs:CombineAllReduces (Results 1 – 2 of 2) sorted by relevance
49 Status CombineAllReduces(absl::Span<HloInstruction* const> to_combine) { in CombineAllReduces() function457 TF_RETURN_IF_ERROR(CombineAllReduces(to_combine)); in Run()
114 TEST_F(AllReduceCombinerTest, CombineAllReduces) { in TEST_F() argument