Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_fusible_test.cc397 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
423 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
445 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion, *reduce)); in TEST_F()
468 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion, *div)); in TEST_F()
502 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
537 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
555 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*reduce, *exp)); in TEST_F()
573 EXPECT_FALSE(ShapesCompatibleForMultiOutputFusion(*reduce, *exp)); in TEST_F()
611 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
641 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
[all …]
H A Dgpu_fusible.cc126 bool ShapesCompatibleForMultiOutputFusion(const HloInstruction& instr1, in ShapesCompatibleForMultiOutputFusion() function
285 if (!ShapesCompatibleForMultiOutputFusion(producer, consumer)) { in IsProducerConsumerMultiOutputFusible()
H A Dgpu_fusible.h103 bool ShapesCompatibleForMultiOutputFusion(const HloInstruction& instr1,
H A Dhorizontal_input_fusion.cc140 if (ShapesCompatibleForMultiOutputFusion(*fusion_anchor, *fused) && in Run()
H A Dmulti_output_fusion.cc224 !ShapesCompatibleForMultiOutputFusion(*(*i), *(*j)) || in FuseSiblings()