Searched refs:CollectiveAllReduceTest (Results 1 – 1 of 1) sorted by relevance
438 class CollectiveAllReduceTest(multi_worker_test_base.MultiWorkerTestBase, class450 super(CollectiveAllReduceTest, self).setUp()453 CollectiveAllReduceTest.collective_key_base += 100000455 CollectiveAllReduceTest.collective_key_base)465 group_key_start=10 + CollectiveAllReduceTest.collective_key_base)525 super(CollectiveAllReduceTest, self)._assert_mirrored_equal(