Searched refs:AllReduceCombinerTest (Results 1 – 1 of 1) sorted by relevance
111 using AllReduceCombinerTest = HloTestBase; typedef114 TEST_F(AllReduceCombinerTest, CombineAllReduces) { in TEST_F() argument158 TEST_F(AllReduceCombinerTest, CombineCrossReplicaReductionsInGroups) { in TEST_F() argument182 TEST_F(AllReduceCombinerTest, RespectThreshold) { in TEST_F() argument214 TEST_F(AllReduceCombinerTest, NoDependentCombination) { in TEST_F() argument240 TEST_F(AllReduceCombinerTest, GroupAllReduce) { in TEST_F() argument268 TEST_F(AllReduceCombinerTest, DomainPreventsCombining) { in TEST_F() argument306 TEST_F(AllReduceCombinerTest, CombineFromTwoDomainsWithSameMetadata) { in TEST_F() argument349 TEST_F(AllReduceCombinerTest, DoNotCombineCrossShardAndCrosReplicaInSPMD) { in TEST_F() argument379 TEST_F(AllReduceCombinerTest, CrossCoreAllReduce) { in TEST_F() argument[all …]