Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.h102 bool IsFusibleAsMultiOutputFusionRoot(const HloInstruction& instr);
Dmulti_output_fusion.cc119 if (!IsFusibleAsMultiOutputFusionRoot(*consumer)) { in GetProducerConsumerMultiOutputFusionCandidates()
172 if (!IsFusibleAsMultiOutputFusionRoot(*instr)) { in IsSiblingFusionCandidate()
Dgpu_fusible_test.cc724 TEST_F(GpuFusibleTest, IsFusibleAsMultiOutputFusionRoot) { in TEST_F() argument
736 EXPECT_TRUE(IsFusibleAsMultiOutputFusionRoot(*root)); in TEST_F()
767 EXPECT_FALSE(IsFusibleAsMultiOutputFusionRoot(*scatter_inst)); in TEST_F()
Dgpu_fusible.cc303 if (!IsLoopFusible(producer) || !IsFusibleAsMultiOutputFusionRoot(consumer)) { in IsProducerConsumerMultiOutputFusible()
507 bool IsFusibleAsMultiOutputFusionRoot(const HloInstruction& instr) { in IsFusibleAsMultiOutputFusionRoot() function