Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible_test.cc363 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
389 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
411 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion, *reduce)); in TEST_F()
433 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion, *div)); in TEST_F()
465 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
482 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*reduce, *exp)); in TEST_F()
500 EXPECT_FALSE(ShapesCompatibleForMultiOutputFusion(*reduce, *exp)); in TEST_F()
538 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
568 EXPECT_TRUE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
599 EXPECT_FALSE(ShapesCompatibleForMultiOutputFusion(*fusion_1, *fusion_2)); in TEST_F()
[all …]
Dmulti_output_fusion.cc44 return ShapesCompatibleForMultiOutputFusion(*instr1, *instr2); in ShapesCompatibleForFusion()
165 if (!ShapesCompatibleForMultiOutputFusion(*producer, *consumer)) { in DoProducerConsumerMultiOutputFusion()
Dgpu_fusible.h69 bool ShapesCompatibleForMultiOutputFusion(const HloInstruction& instr1,
Dgpu_fusible.cc89 bool ShapesCompatibleForMultiOutputFusion(const HloInstruction& instr1, in ShapesCompatibleForMultiOutputFusion() function