Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc226 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
244 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
263 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
282 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
303 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
325 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
346 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
368 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
Dgpu_fusible.h66 bool IsReduceInputFusion(const HloInstruction& instr);
Dgpu_fusible.cc79 bool IsReduceInputFusion(const HloInstruction& instr) { in IsReduceInputFusion() function
103 return IsReduceInputFusion(instr) || in IsInputFusibleReduction()