Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_simplifier.cc174 for (auto* comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
181 for (auto* comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
188 for (auto* comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
194 for (auto* comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
200 for (auto* comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
206 for (auto* comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dhlo_module.h298 std::vector<HloComputation*> MakeNonfusionComputations() const { in MakeNonfusionComputations() function
299 return MakeNonfusionComputations({}); in MakeNonfusionComputations()
303 std::vector<HloComputation*> MakeNonfusionComputations(
H A Dinstruction_hoister.cc32 module.MakeNonfusionComputations(execution_threads)) { in HoistParameters()
86 module.MakeNonfusionComputations(execution_threads)) { in HoistConstantOperations()
H A Dhlo_schedule_test.cc415 ASSERT_EQ(module->MakeNonfusionComputations({"parallel_thread"}).size(), 1); in TEST_F()
417 module->MakeNonfusionComputations({"parallel_thread"}).front())); in TEST_F()
H A Dhlo_schedule.cc220 module_->MakeNonfusionComputations(execution_threads); in Update()
293 module_->MakeNonfusionComputations({thread_name}); in Verify()
H A Dhlo_ordering.cc464 for (auto* computation : module_->MakeNonfusionComputations()) { in ToStringHelper()
486 for (auto* computation : module->MakeNonfusionComputations()) { in DependencyHloOrdering()
H A Dop_expander_pass.cc33 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dcollectives_schedule_linearizer.cc50 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dzero_sized_hlo_elimination.cc35 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dchange_op_data_type.cc27 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Doptimization_barrier_expander.cc25 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dconditional_canonicalizer.cc52 for (auto* comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dreduce_scatter_decomposer.cc44 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Droot_instruction_sinker.cc59 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dbroadcast_canonicalizer.cc32 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dmemory_space_assignment_utils.cc94 for (const HloComputation* computation : module.MakeNonfusionComputations()) { in HoistConstantOperations()
H A Ddynamic_index_splitter.cc38 module->MakeNonfusionComputations(execution_threads); in Run()
H A Dtree_reduction_rewriter.cc119 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dall_gather_decomposer.cc87 for (auto comp : module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dbatch_dot_simplification.cc117 module->MakeNonfusionComputations(execution_threads)) { in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmulti_output_fusion_test.cc50 for (auto* computation : module->MakeNonfusionComputations()) { in CountMultiOutputFusions()
706 for (auto* computation : module->MakeNonfusionComputations()) { in TEST_F()
H A Dtriangular_solve_rewriter.cc31 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dvariadic_op_splitter.cc97 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dgemm_broadcast_folding_rewriter.cc113 module->MakeNonfusionComputations(execution_threads)) { in Run()
H A Dgpu_reduce_scatter_creator.cc37 module->MakeNonfusionComputations(execution_threads)) { in Run()

1234