Home
last modified time | relevance | path

Searched refs:IsMultiOutputFusion (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmulti_output_fusion_test.cc52 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 …]
Dmulti_output_fusion.cc85 if (instr->IsMultiOutputFusion()) { in FusionPriority()
113 !producer->users()[0]->IsMultiOutputFusion()) { in GetProducerConsumerMultiOutputFusionCandidates()
178 if (instr->IsMultiOutputFusion()) { in IsSiblingFusionCandidate()
Dgpu_fusible.cc114 if (instr.IsMultiOutputFusion()) { in IsReduceInputFusion()
152 if (!instr.IsMultiOutputFusion()) { in GetRealHeroForMultiOutputFusion()
255 if (producer.IsMultiOutputFusion()) { in IsProducerConsumerFusible()
299 if (producer.IsMultiOutputFusion()) { in IsProducerConsumerMultiOutputFusible()
Dhorizontal_input_fusion_test.cc75 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
Dfusion_merger.cc106 if (fusion->IsMultiOutputFusion()) { in GetCurrentBytesTransferred()
Dhorizontal_loop_fusion_test.cc83 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
Dir_emitter_unnested.cc339 const Shape& element_shape = hlo->IsMultiOutputFusion() in ComputeMaxUnrollFactor()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmulti_output_fusion.cc147 if (fused->IsMultiOutputFusion()) { in Fuse()
322 if (!instr->IsMultiOutputFusion()) { in LegalToFuseMainConstraints()
Dhlo_instruction.h1757 const bool IsMultiOutputFusion() const;
Dhlo_instructions.h969 const bool IsMultiOutputFusion() const { in IsMultiOutputFusion() function
Dhlo_instruction.cc4045 const bool HloInstruction::IsMultiOutputFusion() const { in IsMultiOutputFusion() function in xla::HloInstruction
4047 return fusion != nullptr && fusion->IsMultiOutputFusion(); in IsMultiOutputFusion()