Searched refs:LayoutsAreReduceInputFusionFriendly (Results 1 – 3 of 3) 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()
129 LayoutsAreReduceInputFusionFriendly(*loop_fusion, *reduce_fusion)); in TEST_F()
152 EXPECT_FALSE(LayoutsAreReduceInputFusionFriendly(*copy, *reduce)); in TEST_F()
189 LayoutsAreReduceInputFusionFriendly(*loop_fusion, *reduce_fusion)); in TEST_F()
217 EXPECT_TRUE(LayoutsAreReduceInputFusionFriendly(*loop_fusion, *reduce)); in TEST_F()
|
| D | gpu_fusible.cc | 91 bool LayoutsAreReduceInputFusionFriendly(const HloInstruction& producer, in LayoutsAreReduceInputFusionFriendly() function
270 !LayoutsAreReduceInputFusionFriendly(producer, consumer)) { in IsProducerConsumerFusible()
312 if (!LayoutsAreReduceInputFusionFriendly(producer, consumer)) { in IsProducerConsumerMultiOutputFusible()
|
| D | gpu_fusible.h | 39 bool LayoutsAreReduceInputFusionFriendly(const HloInstruction& producer,
|