Searched refs:AllReduceCount (Results 1 – 1 of 1) sorted by relevance
44 int64 AllReduceCount(const HloModule& module) { in AllReduceCount() function126 ASSERT_EQ(AllReduceCount(*module), inputs.size()); in TEST_F()128 ASSERT_EQ(AllReduceCount(*module), 1); in TEST_F()174 ASSERT_EQ(AllReduceCount(*module), inputs.size()); in TEST_F()176 ASSERT_EQ(AllReduceCount(*module), 3) in TEST_F()196 ASSERT_EQ(AllReduceCount(*module), inputs.size()); in TEST_F()198 EXPECT_EQ(AllReduceCount(*module), inputs.size()); in TEST_F()206 ASSERT_EQ(AllReduceCount(*module), inputs.size()); in TEST_F()208 EXPECT_EQ(AllReduceCount(*module), 1); in TEST_F()233 ASSERT_EQ(AllReduceCount(*module), 2); in TEST_F()[all …]