Home
last modified time | relevance | path

Searched refs:MakeNonfusionComputations (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dop_expander_pass.cc30 for (HloComputation* computation : module->MakeNonfusionComputations()) { in Run()
Dzero_sized_hlo_elimination.cc32 for (HloComputation* comp : module->MakeNonfusionComputations()) { in Run()
Dconditional_canonicalizer.cc50 for (auto* comp : module->MakeNonfusionComputations()) { in Run()
Droot_instruction_sinker.cc56 for (HloComputation* computation : module->MakeNonfusionComputations()) { in Run()
Dhlo_constant_folding.cc76 for (auto* computation : module->MakeNonfusionComputations()) { in Run()
Ddynamic_index_splitter.cc36 module->MakeNonfusionComputations(); in Run()
Dhlo_schedule.cc205 module_->MakeNonfusionComputations(); in Update()
242 module_->MakeNonfusionComputations(); in Verify()
Dtree_reduction_rewriter.cc102 for (const auto &computation : module->MakeNonfusionComputations()) { in Run()
Dmemory_space_propagation.cc26 for (HloComputation* computation : module->MakeNonfusionComputations()) { in Run()
Dbatch_dot_simplification.cc115 for (HloComputation* computation : module->MakeNonfusionComputations()) { in Run()
Dwhile_loop_constant_sinking.cc114 for (auto* comp : module->MakeNonfusionComputations()) { in Run()
Dloop_schedule_linearizer.cc148 for (HloComputation* computation : module->MakeNonfusionComputations()) { in Run()
Dhlo_ordering.cc455 for (auto* computation : module_->MakeNonfusionComputations()) { in ToStringHelper()
477 for (auto* computation : module->MakeNonfusionComputations()) { in DependencyHloOrdering()
Dsort_simplifier.cc144 for (auto* comp : module->MakeNonfusionComputations()) { in Run()
Dlogical_buffer_analysis.cc65 for (auto* computation : module_->MakeNonfusionComputations()) { in Analyze()
Dall_gather_decomposer.cc142 for (auto comp : module->MakeNonfusionComputations()) { in Run()
Dbfloat16_conversion_folding.cc260 for (auto* comp : module->MakeNonfusionComputations()) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc263 for (HloComputation* computation : module->MakeNonfusionComputations()) { in RunFuseBiasSideActivation()
394 for (HloComputation* computation : module->MakeNonfusionComputations()) { in RunFuseClamp()
462 for (HloComputation* computation : module->MakeNonfusionComputations()) { in RunFuseConvertToFloat()
Dvariadic_op_splitter.cc94 for (HloComputation* comp : module->MakeNonfusionComputations()) { in Run()
Dhorizontal_input_fusion.cc159 for (auto* comp : module->MakeNonfusionComputations()) { in Run()
Dgemm_rewriter.cc135 for (HloComputation *computation : module->MakeNonfusionComputations()) { in Run()
Dmulti_output_fusion_test.cc50 for (auto* computation : module->MakeNonfusionComputations()) { in CountMultiOutputFusions()
706 for (auto* computation : module->MakeNonfusionComputations()) { in TEST_F()
Dcublas_gemm_pad_for_tensor_cores.cc166 for (HloComputation* comp : module->MakeNonfusionComputations()) { in Run()
Dcusolver_rewriter.cc180 for (HloComputation* computation : module->MakeNonfusionComputations()) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc254 for (auto* computation : module->MakeNonfusionComputations()) { in ComputeTargetParallelTasks()

123