Home
last modified time | relevance | path

Searched refs:IsReduceInputFusion (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc233 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
251 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
270 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
289 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
310 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
332 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
353 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
375 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
Dgpu_fusible.h49 bool IsReduceInputFusion(const HloInstruction& instr);
Dhorizontal_input_fusion.cc85 if (IsReduceInputFusion(*predecessor) && in FindAndSortFusionCandidates()
Dgpu_fusible.cc113 bool IsReduceInputFusion(const HloInstruction& instr) { in IsReduceInputFusion() function
142 return IsReduceInputFusion(instr) || in IsInputFusibleReduction()