Home
last modified time | relevance | path

Searched defs:execution_threads (Results 1 – 25 of 132) sorted by relevance

123456

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_pass_fix.h43 execution_threads) override { in RunOnChangedComputations()
54 const absl::flat_hash_set<absl::string_view>& in Run()
64 execution_threads) override { in RunOnModuleGroup()
89 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunToFixPoint()
112 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunOnChangedComputationsOnce()
Dhlo_domain_verifier.cc47 const absl::flat_hash_set<absl::string_view>& execution_threads) { in PopulateDomainKinds()
69 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
89 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dinstruction_hoister.cc27 const absl::flat_hash_set<absl::string_view>& execution_threads) { in HoistParameters()
81 const absl::flat_hash_set<absl::string_view>& execution_threads) { in HoistConstantOperations()
136 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dhlo_pass_interface.h102 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunOnChangedComputations()
152 execution_threads) override { in RunOnModuleGroup()
175 const absl::flat_hash_set<absl::string_view>& in Run()
Dhlo_pass_pipeline.cc128 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunInvariantCheckers()
151 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunPassesInternal()
272 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
284 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunOnModuleGroup()
Dhlo_domain_remover.cc62 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
125 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dhlo_domain_isolator.cc63 const absl::flat_hash_set<absl::string_view>& execution_threads, in RunInternal()
130 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dhlo_module_dce.cc42 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunWhileDCE()
110 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dhlo_pass_pipeline_test.cc53 const absl::flat_hash_set<absl::string_view>& in Run()
76 const absl::flat_hash_set<absl::string_view>& in Run()
98 execution_threads) override { in RunOnModuleGroup()
122 const absl::flat_hash_set<absl::string_view>& in Run()
Dasync_op_canonicalizer.cc29 const absl::flat_hash_set<absl::string_view>& execution_threads, in CreateAsyncGroups()
95 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dhlo_pass_pipeline.h139 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunHelper()
146 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RunHelper()
Dhlo_dce.cc129 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RecursivelyRemoveDeadComputations()
170 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Ddespecializer.cc34 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dzero_sized_hlo_elimination.cc32 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dsharding_remover.cc36 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dop_expander_pass.cc30 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Ddespecializer.h54 const absl::flat_hash_set<absl::string_view>& in Run()
Dhlo_memory_scheduler.cc535 int64_t* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()
667 const absl::flat_hash_set<absl::string_view>& execution_threads, in DefaultModuleScheduler()
726 const absl::flat_hash_set<absl::string_view>& execution_threads, in ScheduleModule()
765 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
775 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
797 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dcollectives_schedule_linearizer.cc47 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dwhile_loop_trip_count_annotator.cc24 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dhlo_subcomputation_unification.cc22 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.h35 const absl::flat_hash_set<absl::string_view>& in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion.h36 const absl::flat_hash_set<absl::string_view>& in Run()
Dalias_passthrough_params.cc26 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
Dgpu_sanitize_constant_names.cc34 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()

123456