Searched refs:crs0 (Results 1 – 1 of 1) sorted by relevance
247 auto crs0 = b.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F() local257 b.AddInstruction(HloInstruction::CreateTuple({crs0, crs1})); in TEST_F()469 auto crs0 = op::AllReduce(op::Parameter(0), op::AllReduce(op::Parameter(1))); in TEST_F() local470 auto add = op::Add(op::AllReduce(op::GetTupleElement(crs0, 0)), in TEST_F()471 op::GetTupleElement(crs0, 1)); in TEST_F()472 auto crs1 = op::AllReduce(add, op::GetTupleElement(crs0)); in TEST_F()