Searched refs:LayoutsAreReduceInputFusionFriendly (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible_test.cc | 53 EXPECT_TRUE(LayoutsAreReduceInputFusionFriendly(*exp, *reduce)); in TEST_F()
91 LayoutsAreReduceInputFusionFriendly(*loop_fusion, *reduce_fusion)); in TEST_F()
114 EXPECT_FALSE(LayoutsAreReduceInputFusionFriendly(*copy, *reduce)); in TEST_F()
151 LayoutsAreReduceInputFusionFriendly(*loop_fusion, *reduce_fusion)); in TEST_F()
178 EXPECT_TRUE(LayoutsAreReduceInputFusionFriendly(*loop_fusion, *reduce)); in TEST_F()
|
| D | gpu_fusible.h | 42 bool LayoutsAreReduceInputFusionFriendly(const HloInstruction& producer,
|
| D | gpu_fusible.cc | 44 bool LayoutsAreReduceInputFusionFriendly(const HloInstruction& producer, in LayoutsAreReduceInputFusionFriendly() function
|
| D | fusion_merger.cc | 211 LayoutsAreReduceInputFusionFriendly(*fusion, *user))); in HandleFusion()
|
| D | multi_output_fusion.cc | 169 if (!LayoutsAreReduceInputFusionFriendly(*producer, *consumer)) { in DoProducerConsumerMultiOutputFusion()
|
| D | instruction_fusion.cc | 217 !LayoutsAreReduceInputFusionFriendly(*producer, *consumer)) { in ShouldFuseInexpensiveChecks()
|