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.cc194 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
211 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
230 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
249 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
270 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
292 EXPECT_TRUE(IsReduceInputFusion(*reduce)); in TEST_F()
313 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
335 EXPECT_FALSE(IsReduceInputFusion(*reduce)); in TEST_F()
Dgpu_fusible.cc63 bool IsReduceInputFusion(const HloInstruction& instr) { in IsReduceInputFusion() function
86 return IsReduceInputFusion(instr) || IsReductionToVector(instr); in IsInputFusibleReduction()
Dgpu_fusible.h52 bool IsReduceInputFusion(const HloInstruction& instr);
Dfusion_merger.cc210 (IsReduceInputFusion(*user) && in HandleFusion()