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.h119 bool IsFusibleAsMultiOutputFusionRoot(const HloInstruction& instr);
Dmulti_output_fusion.cc125 if (!IsFusibleAsMultiOutputFusionRoot(*consumer)) { in GetProducerConsumerMultiOutputFusionCandidates()
180 if (!IsFusibleAsMultiOutputFusionRoot(*instr)) { in IsSiblingFusionCandidate()
Dgpu_fusible_test.cc716 TEST_F(GpuFusibleTest, IsFusibleAsMultiOutputFusionRoot) { in TEST_F() argument
728 EXPECT_TRUE(IsFusibleAsMultiOutputFusionRoot(*root)); in TEST_F()
759 EXPECT_FALSE(IsFusibleAsMultiOutputFusionRoot(*scatter_inst)); in TEST_F()
Dgpu_fusible.cc279 if (!IsLoopFusible(producer) || !IsFusibleAsMultiOutputFusionRoot(consumer)) { in IsProducerConsumerMultiOutputFusible()
557 bool IsFusibleAsMultiOutputFusionRoot(const HloInstruction& instr) { in IsFusibleAsMultiOutputFusionRoot() function