Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc404 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
430 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
452 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion, *reduce)); in TEST_F()
475 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion, *div)); in TEST_F()
509 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
544 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
562 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*reduce, *exp)); in TEST_F()
580 EXPECT_FALSE(ShapesCompatibleForMultiOutputFusion(*reduce, *exp)); in TEST_F()
618 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
648 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
[all …]
Dgpu_fusible.h86 bool ShapesCompatibleForMultiOutputFusion(const HloInstruction& instr1,
Dhorizontal_input_fusion.cc130 if (ShapesCompatibleForMultiOutputFusion(*fusion_anchor, *fused) && in Run()
Dgpu_fusible.cc165 bool ShapesCompatibleForMultiOutputFusion(const HloInstruction& instr1, in ShapesCompatibleForMultiOutputFusion() function
309 if (!ShapesCompatibleForMultiOutputFusion(producer, consumer)) { in IsProducerConsumerMultiOutputFusible()
Dmulti_output_fusion.cc215 !ShapesCompatibleForMultiOutputFusion(*(*i), *(*j)) || in FuseSiblings()