Home
last modified time | relevance | path

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

123456

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A Dbfloat16_propagation.cc697 const absl::flat_hash_set<absl::string_view>& execution_threads) { in ResolveInconsistencyOfAliasingBuffers()
712 const absl::flat_hash_set<absl::string_view>& execution_threads) { in ResolveInconsistentFusions()
775 const absl::flat_hash_set<absl::string_view>& execution_threads) { in ResolveConvertedConstants()
807 const absl::flat_hash_set<absl::string_view>& execution_threads) { in SkipNoopConversions()
835 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A Dcopy_insertion.cc1815 const absl::flat_hash_set<absl::string_view>& execution_threads) { in AddCopiesToResolveInterference()
1852 const absl::flat_hash_set<absl::string_view>& execution_threads) { in AddSpecialCaseCopies()
1859 const absl::flat_hash_set<absl::string_view>& execution_threads, in AddSpecialCaseCopies()
2012 const absl::flat_hash_set<absl::string_view>& execution_threads) { in GetNumExistingCopies()
2026 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RemoveUnnecessaryCopies()
2091 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A Dtopk_rewriter.cc187 const absl::flat_hash_set<absl::string_view>& execution_threads) { in TransformToCustomCall()
276 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A Dsharding_propagation.cc805 const absl::flat_hash_set<absl::string_view>& execution_threads) { in AssignShardingMetadata()
829 const absl::flat_hash_set<absl::string_view>& execution_threads) { in RemoveShardingMetadata()
1246 const absl::flat_hash_set<absl::string_view>& execution_threads, in ProcessShardingInstruction()
2279 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A Ddespecializer.cc34 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A Dzero_sized_hlo_elimination.cc32 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A Dinstruction_fusion.cc486 const absl::flat_hash_set<absl::string_view>& execution_threads) { in GetFusionComputations()
498 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A Ddespecializer.h54 const absl::flat_hash_set<absl::string_view>& in Run()
H A Dsharding_remover.cc36 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
H A Dop_expander_pass.cc30 const absl::flat_hash_set<absl::string_view>& execution_threads) { in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion.h35 const absl::flat_hash_set<absl::string_view>& in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dinstruction_fusion.h36 const absl::flat_hash_set<absl::string_view>& in Run()

123456