Searched refs:all_reduce_analysis (Results 1 – 1 of 1) sorted by relevance
990 HloCostAnalysis all_reduce_analysis(ShapeSize); in TEST_F() local991 ASSERT_IS_OK(all_reduce->Accept(&all_reduce_analysis)); in TEST_F()993 EXPECT_EQ(all_reduce_analysis.bytes_accessed(*all_reduce), in TEST_F()995 EXPECT_EQ(all_reduce_analysis.operand_bytes_accessed(*all_reduce, 0), in TEST_F()997 EXPECT_EQ(all_reduce_analysis.operand_bytes_accessed(*all_reduce, 1), in TEST_F()999 EXPECT_EQ(all_reduce_analysis.output_bytes_accessed(*all_reduce), in TEST_F()