Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc234 EXPECT_TRUE(IsInputFusibleReduction(*reduce)); in TEST_F()
252 EXPECT_FALSE(IsInputFusibleReduction(*reduce)); in TEST_F()
271 EXPECT_TRUE(IsInputFusibleReduction(*reduce)); in TEST_F()
290 EXPECT_FALSE(IsInputFusibleReduction(*reduce)); in TEST_F()
311 EXPECT_TRUE(IsInputFusibleReduction(*reduce)); in TEST_F()
333 EXPECT_TRUE(IsInputFusibleReduction(*reduce)); in TEST_F()
354 EXPECT_FALSE(IsInputFusibleReduction(*reduce)); in TEST_F()
376 EXPECT_FALSE(IsInputFusibleReduction(*reduce)); in TEST_F()
Dgpu_fusible.cc136 bool IsInputFusibleReduction(const HloInstruction& instr) { in IsInputFusibleReduction() function
207 (IsInputFusibleReduction(instr) || IsInputFusibleScatter(instr)); in IsInputFusible()
269 if (IsInputFusibleReduction(consumer) && in IsProducerConsumerFusible()
514 (IsInputFusibleReduction(instr) || in IsFusibleAsMultiOutputFusionRoot()
Dgpu_fusible.h53 bool IsInputFusibleReduction(const HloInstruction& instr);