Home
last modified time | relevance | path

Searched refs:IsProducerConsumerFusible (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.h92 bool IsProducerConsumerFusible(const HloInstruction& producer,
Dgpu_fusible_test.cc940 EXPECT_FALSE(IsProducerConsumerFusible(*producer, *consumer)); in TEST_F()
941 EXPECT_FALSE(IsProducerConsumerFusible(*producer2, *consumer2)); in TEST_F()
965 EXPECT_FALSE(IsProducerConsumerFusible(*producer, *consumer)); in TEST_F()
981 EXPECT_TRUE(IsProducerConsumerFusible(*producer, *consumer)); in TEST_F()
Dinstruction_fusion.cc75 if (!IsProducerConsumerFusible(*producer, *consumer) || in ShouldFuseInexpensiveChecks()
Dfusion_merger.cc205 return IsProducerConsumerFusible(*fusion, *user) && in HandleFusion()
Dgpu_fusible.cc241 bool IsProducerConsumerFusible(const HloInstruction& producer, in IsProducerConsumerFusible() function