Searched refs:IsMultiOutputFusion (Results 1 – 11 of 11) sorted by relevance
52 if (instr->IsMultiOutputFusion()) { in CountMultiOutputFusions()84 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()172 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()205 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()268 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()313 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()377 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()433 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()461 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()508 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()[all …]
114 if (instr.IsMultiOutputFusion()) { in IsReduceInputFusion()152 if (!instr.IsMultiOutputFusion()) { in GetRealHeroForMultiOutputFusion()255 if (producer.IsMultiOutputFusion()) { in IsProducerConsumerFusible()299 if (producer.IsMultiOutputFusion()) { in IsProducerConsumerMultiOutputFusible()
85 if (instr->IsMultiOutputFusion()) { in FusionPriority()113 !producer->users()[0]->IsMultiOutputFusion()) { in GetProducerConsumerMultiOutputFusionCandidates()178 if (instr->IsMultiOutputFusion()) { in IsSiblingFusionCandidate()
75 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
106 if (fusion->IsMultiOutputFusion()) { in GetCurrentBytesTransferred()
83 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
339 const Shape& element_shape = hlo->IsMultiOutputFusion() in ComputeMaxUnrollFactor()
147 if (fused->IsMultiOutputFusion()) { in Fuse()322 if (!instr->IsMultiOutputFusion()) { in LegalToFuseMainConstraints()
1757 const bool IsMultiOutputFusion() const;
969 const bool IsMultiOutputFusion() const { in IsMultiOutputFusion() function
4045 const bool HloInstruction::IsMultiOutputFusion() const { in IsMultiOutputFusion() function in xla::HloInstruction4047 return fusion != nullptr && fusion->IsMultiOutputFusion(); in IsMultiOutputFusion()