Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmulti_output_fusion_test.cc71 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
159 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
192 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
254 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
298 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
354 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
402 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
429 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
470 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
537 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
Dgpu_fusible.cc64 if (instr.IsMultiOutputFusion()) { in IsReduceInputFusion()
97 if (instr->IsMultiOutputFusion()) { in ShapesCompatibleForMultiOutputFusion()
Dfusion_merger.cc105 if (fusion->IsMultiOutputFusion()) { in GetCurrentBytesTransferred()
199 if (fusion->IsMultiOutputFusion()) { in HandleFusion()
Dir_emitter_nested.cc121 if (hlo.IsMultiOutputFusion()) { in EmitTargetElementLoop()
Dir_emitter_unnested.cc271 const Shape& element_shape = hlo->IsMultiOutputFusion() in ComputeMaxUnrollFactor()
1918 CHECK(hlo->IsMultiOutputFusion()) in BuildInitializerThunk()
2180 const Shape& element_shape = hlo.IsMultiOutputFusion() in EmitTargetElementLoopInThunk()
2190 if (!hlo.IsMultiOutputFusion()) { in EmitTargetElementLoopInThunk()
2506 if (hlo->IsMultiOutputFusion()) { in EmitTileElementForFusion()
3096 if (!reduction_info && unnested_hlo->IsMultiOutputFusion()) { in EmitKernel()
3372 const Shape& output_shape = hlo->IsMultiOutputFusion() in CheckAndEmitHloWithTile021()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmulti_output_fusion.cc122 if (fused->IsMultiOutputFusion()) { in Fuse()
235 if (!instr->IsMultiOutputFusion()) { in LegalToFuse()
Dhlo_instruction.h1477 const bool IsMultiOutputFusion() const;
Dhlo_instructions.h785 const bool IsMultiOutputFusion() const { in IsMultiOutputFusion() function
Dhlo_instruction.cc3350 const bool HloInstruction::IsMultiOutputFusion() const { in IsMultiOutputFusion() function in xla::HloInstruction
3352 return fusion != nullptr && fusion->IsMultiOutputFusion(); in IsMultiOutputFusion()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2957 if (target_op->IsMultiOutputFusion()) { in EmitTargetElementLoop()