Searched refs:fusion_analysis (Results 1 – 1 of 1) sorted by relevance
480 HloCostAnalysis fusion_analysis(ShapeSize); in TEST_F() local481 fusion_analysis.set_flops_per_second(16 * (i == 1 ? 1 / 2.0 : 1.0)); in TEST_F()482 fusion_analysis.set_transcendentals_per_second(4 * in TEST_F()484 fusion_analysis.set_bytes_per_second(64 * (i == 3 ? 1 / 8.0 : 1.0)); in TEST_F()485 ASSERT_IS_OK(fusion->Accept(&fusion_analysis)); in TEST_F()487 EXPECT_EQ(fusion_analysis.flop_count(), 16); in TEST_F()488 EXPECT_EQ(fusion_analysis.transcendental_count(), 4); in TEST_F()491 EXPECT_EQ(fusion_analysis.bytes_accessed(), bytes_accessed); in TEST_F()493 EXPECT_EQ(fusion_analysis.optimal_seconds(), 1 << i); in TEST_F()519 HloCostAnalysis fusion_analysis(ShapeSize); in TEST_F() local[all …]