Home
last modified time | relevance | path

Searched refs:IsLoopFusion (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc180 if (consumer->IsLoopFusion()) { in ShouldFuse()
Dparallel_task_assignment.cc156 if (instruction->IsElementwise() || instruction->IsLoopFusion() || in GetTargetParallelTaskCount()
Dir_emitter.cc2186 } else if (fusion->IsLoopFusion()) { in HandleFusion()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_merger.cc262 if (!fusion->IsLoopFusion()) { in HandleFusion()
Dmulti_output_fusion.cc66 instr1->IsLoopFusion())) { in LegalToFuse()
Dgpu_fusible.cc565 instr.IsLoopFusion() || // TODO(b/130013493): Use IsLoopFusible here. in IsFusibleAsMultiOutputFusionRoot()
Dhorizontal_loop_fusion.cc109 if (!instr.IsLoopFusion()) { in IsFusibleCandidate()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.cc47 if (instr->IsLoopFusion()) { in MayBeImplementedAsInPlaceDynamicUpdateSlice()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc723 } else if (user->IsLoopFusion()) { in DoesNotUseOperandBuffer()
Dhlo_dataflow_analysis.cc1631 if (user->IsLoopFusion()) { in DoesNotUseOperandBuffer()
1906 if (user->IsLoopFusion() || user->IsInputFusion()) { in CanShareOperandBufferWithUser()
Dhlo_instruction.h1536 bool IsLoopFusion() const;
Dhlo_dataflow_analysis_test.cc2763 return fusion->IsLoopFusion(); in TEST_F()
Dhlo_instruction.cc3427 bool HloInstruction::IsLoopFusion() const { in IsLoopFusion() function in xla::HloInstruction