Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc784 EXPECT_TRUE(IsProducerConsumerMultiOutputFusible(*producer, *consumer)); in TEST_F()
808 EXPECT_TRUE(IsProducerConsumerMultiOutputFusible(*producer, *consumer)); in TEST_F()
847 EXPECT_TRUE(IsProducerConsumerMultiOutputFusible(*producer, *consumer)); in TEST_F()
880 EXPECT_FALSE(IsProducerConsumerMultiOutputFusible(*producer, *consumer)); in TEST_F()
911 EXPECT_FALSE(IsProducerConsumerMultiOutputFusible(*producer, *consumer)); in TEST_F()
Dgpu_fusible.h98 bool IsProducerConsumerMultiOutputFusible(const HloInstruction& producer,
Dmulti_output_fusion.cc124 if (!IsProducerConsumerMultiOutputFusible(*producer, *consumer)) { in GetProducerConsumerMultiOutputFusionCandidates()
Dgpu_fusible.cc296 bool IsProducerConsumerMultiOutputFusible(const HloInstruction& producer, in IsProducerConsumerMultiOutputFusible() function