Home
last modified time | relevance | path

Searched refs:alias_analysis (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dside_effect_analysis.cc70 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis) { in FindAccessedResources() argument
75 if (alias_analysis.IsUnknownResource(operand)) { in FindAccessedResources()
79 const auto& ids = alias_analysis.GetResourceUniqueIds(operand); in FindAccessedResources()
83 if (alias_analysis.IsUnknownResource(result)) { in FindAccessedResources()
87 const auto& ids = alias_analysis.GetResourceUniqueIds(result); in FindAccessedResources()
207 const ResourceAliasAnalysis::Info& alias_analysis) { in OpIsDeclaration() argument
209 !FindAccessedResources(op, alias_analysis).empty(); in OpIsDeclaration()
219 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis, in GetResourceIdsByValue() argument
234 if (alias_analysis.IsUnknownResource(value)) return false; in GetResourceIdsByValue()
236 const auto& ids = alias_analysis.GetResourceUniqueIds(value); in GetResourceIdsByValue()
[all …]
Dside_effect_analysis.h43 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo() argument
44 AnalyzeFunction(func_op, alias_analysis); in SideEffectAnalysisInfo()
49 Region* region, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo() argument
50 AnalyzeRegion(region, alias_analysis); in SideEffectAnalysisInfo()
73 const TF::ResourceAliasAnalysis::Info& alias_analysis);
77 const TF::ResourceAliasAnalysis::Info& alias_analysis);
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc416 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper() argument
425 return algorithm(computation, points_to_analysis, alias_analysis, in ScheduleComputationHelper()
429 return DefaultMemoryScheduler(computation, points_to_analysis, alias_analysis, in ScheduleComputationHelper()
439 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler() argument
513 *computation, sequence, alias_analysis, size_function, in DFSMemoryScheduler()
524 const HloAliasAnalysis& alias_analysis, in ComputationSchedulerToModuleScheduler()
534 computation, points_to_analysis, alias_analysis, size_func, in ComputationSchedulerToModuleScheduler()
551 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler() argument
565 *computation, sequence, alias_analysis, size_function, in ListMemoryScheduler()
574 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler() argument
[all …]
Dhlo_memory_scheduler.h73 const HloAliasAnalysis& alias_analysis,
83 const HloAliasAnalysis& alias_analysis,
93 const HloAliasAnalysis& alias_analysis,
106 const HloAliasAnalysis& alias_analysis,
114 const HloAliasAnalysis& alias_analysis,
Dheap_simulator.cc50 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in MinimumMemoryForModule()
62 schedule, *alias_analysis, size_function)); in MinimumMemoryForModule()
69 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation() argument
77 sequence, alias_analysis, size_function, HeapSimulator::Options(), in MinimumMemoryForComputation()
84 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation() argument
91 sequence, alias_analysis, size_function, schedule, in MinimumMemoryForComputation()
99 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run() argument
107 HloLiveRange::Run(schedule, alias_analysis, entry_computation)); in Run()
109 instruction_sequence, alias_analysis, in Run()
119 const HloAliasAnalysis& alias_analysis, in Run() argument
[all …]
Dhlo_alias_analysis.cc531 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run() local
532 TF_ASSIGN_OR_RETURN(alias_analysis->dataflow_analysis_, in Run()
537 BufferValueMap buffer_map(module, alias_analysis->dataflow_analysis()); in Run()
545 alias_analysis->buffers_.reserve(sorted_buffer_numbers.size()); in Run()
552 alias_analysis->buffers_.emplace_back(next_id++, sorted_values); in Run()
554 alias_analysis->value_to_buffer_[value] = in Run()
555 &alias_analysis->buffers_.back(); in Run()
559 TF_DCHECK_OK(alias_analysis->Verify()); in Run()
565 alias_analysis->ComputeBuffersAt(root, index)) { in Run()
566 alias_analysis->live_out_buffers_.insert(buffer); in Run()
[all …]
Dloop_schedule_linearizer.cc71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites() argument
72 HloDataflowAnalysis& dataflow = alias_analysis.dataflow_analysis(); in AddControlEdgesForLoopWrites()
144 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in Run()
153 AddControlEdgesForLoopWrites(instruction, *alias_analysis); in Run()
Dbuffer_assignment.cc591 if (alias_analysis().BufferLivesOut(buffer)) { in AddAssignment()
605 if (alias_analysis().ValueLivesOut(hlo_value)) { in AddAssignment()
890 alias_analysis().GetBufferContainingValue(*value).values()) { in ToProto()
1042 if (assignment->alias_analysis().BufferLivesOut(hlo_buffer)) { in MaybeAssignBuffer()
1099 if (assignment->alias_analysis().BufferLivesOut(hlo_buffer) && in MaybeAssignBuffer()
1210 !assignment->alias_analysis().BufferLivesOut(*hlo_buffer)) { in AssignSingleHloBuffer()
1267 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations() local
1269 for (const HloBuffer& buffer : alias_analysis.buffers()) { in AssignBuffersForComputations()
1323 sorted_buffers, [&post_order_position, &alias_analysis, assignment]( in AssignBuffersForComputations()
1332 const bool a_live_out = alias_analysis.BufferLivesOut(*a); in AssignBuffersForComputations()
[all …]
Dhlo_live_range.h46 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis,
107 const HloAliasAnalysis& alias_analysis, in HloLiveRange() argument
110 alias_analysis_(alias_analysis), in HloLiveRange()
Dheap_simulator.h123 const HloAliasAnalysis& alias_analysis,
130 const HloAliasAnalysis& alias_analysis,
146 const HloAliasAnalysis& alias_analysis,
158 const HloAliasAnalysis& alias_analysis,
170 const HloAliasAnalysis& alias_analysis,
187 const HloAliasAnalysis& alias_analysis,
Dcopy_insertion.cc284 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile() argument
290 if (!IndicesToCopyForWhile(alias_analysis.dataflow_analysis(), xla_while, in AddCopiesForWhile()
342 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation() argument
1136 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover() argument
1138 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) { in CopyRemover()
1144 for (const HloBuffer& buffer : alias_analysis.buffers()) { in CopyRemover()
1747 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional() argument
1752 if (!IndicesToCopyForConditional(alias_analysis.dataflow_analysis(), in AddCopiesForConditional()
1777 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddCopiesToResolveInterference()
1784 TF_RETURN_IF_ERROR(AddCopiesForWhile(*alias_analysis, instruction)); in AddCopiesToResolveInterference()
[all …]
Dbuffer_assignment.h463 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis() function
503 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment() argument
509 alias_analysis_(std::move(alias_analysis)), in BufferAssignment()
593 return [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in DefaultColorer()
594 for (HloValue* value : alias_analysis->dataflow_analysis().values()) { in DefaultColorer()
Dmemory_space_assignment.h180 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis() argument
185 alias_analysis_(std::move(alias_analysis)), in MemorySpaceAssignmentCostAnalysis()
820 const HloAliasAnalysis& alias_analysis, const Options& options);
837 const HloAliasAnalysis& alias_analysis);
843 const HloAliasAnalysis& alias_analysis);
1091 const Options& options, const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap() argument
1096 alias_analysis_(alias_analysis), in AlternateMemoryBestFitHeap()
1140 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis() function
Dcopy_insertion.h92 virtual Status AddCopiesForConditional(const HloAliasAnalysis& alias_analysis,
Dhlo_live_range.cc25 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run() argument
28 new HloLiveRange(schedule, alias_analysis, module_scoped_analysis)); in Run()
Dmemory_space_assignment.cc94 FindCrossProgramPrefetchCandidate(const HloAliasAnalysis& alias_analysis, in FindCrossProgramPrefetchCandidate() argument
98 for (const HloBuffer& buffer : alias_analysis.buffers()) { in FindCrossProgramPrefetchCandidate()
151 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
153 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
157 cost_analysis, options, std::move(alias_analysis), in Create()
2881 const HloAliasAnalysis& alias_analysis, in Run() argument
2891 alias_analysis); in Run()
2897 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment() argument
2898 TF_RETURN_IF_ERROR(FindAllocationSequence(hlo_live_range, alias_analysis)); in RunMemorySpaceAssignment()
2930 const HloAliasAnalysis& alias_analysis) { in FindAllocationSequence() argument
[all …]
Dhlo_memory_scheduler_test.cc46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation() local
55 *computation, sequence, *alias_analysis, size_function) in PeakMemoryUseOfEntryComputation()
Dhlo_module_group_metadata.h211 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis() function
Dhlo_module_group_metadata.cc153 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in Build()
155 alias_analyses_[module] = std::move(alias_analysis); in Build()
Dmemory_space_assignment_test.cc65 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis() local
158 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace() local
160 HloLiveRange::Run(module->schedule(), *alias_analysis, in AssignMemorySpace()
165 MemorySpaceAssignment::Run(module, *hlo_live_range, *alias_analysis, in AssignMemorySpace()
259 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); local
260 HloBuffer& buffer = alias_analysis->GetUniqueBufferAt(instruction, index);
341 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
343 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
347 cost_analysis, options, std::move(alias_analysis), in Create()
403 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis() argument
[all …]
Dbuffer_assignment_test.cc587 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F() argument
590 id < alias_analysis->dataflow_analysis().values().size(); id++) { in TEST_F()
591 auto& value = alias_analysis->dataflow_analysis().GetValue(id); in TEST_F()
653 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F() argument
655 id < alias_analysis->dataflow_analysis().values().size(); id++) { in TEST_F()
656 auto& value = alias_analysis->dataflow_analysis().GetValue(id); in TEST_F()
657 auto& buffer = alias_analysis->GetBufferContainingValue(value); in TEST_F()
Dheap_simulator_test.cc211 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F() local
218 *alias_analysis, size_fn, &memory_by_computation) in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
DBUILD33 name = "alias_analysis",
34 srcs = ["alias_analysis.cc"],
35 hdrs = ["alias_analysis.h"],
55 ":alias_analysis",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_device_inference.cc78 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult() argument
79 : alias_analysis_(alias_analysis) {} in PerFunctionResult()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
DBUILD108 "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",

12