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.cc69 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis) { in FindAccessedResources() argument
73 if (alias_analysis.IsUnknownResource(operand)) return UnknownResourceSet(); in FindAccessedResources()
74 const auto& ids = alias_analysis.GetResourceUniqueIds(operand); in FindAccessedResources()
78 if (alias_analysis.IsUnknownResource(result)) return UnknownResourceSet(); in FindAccessedResources()
79 const auto& ids = alias_analysis.GetResourceUniqueIds(result); in FindAccessedResources()
188 const ResourceAliasAnalysis::Info& alias_analysis) { in OpIsDeclaration() argument
190 !FindAccessedResources(op, alias_analysis).empty(); in OpIsDeclaration()
200 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis, in GetResourceIdsByValue() argument
207 if (alias_analysis.IsUnknownResource(value)) return false; in GetResourceIdsByValue()
208 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()
428 return DefaultMemoryScheduler(computation, points_to_analysis, alias_analysis, in ScheduleComputationHelper()
438 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler() argument
509 *computation, sequence, alias_analysis, size_function, in DFSMemoryScheduler()
519 const HloAliasAnalysis& alias_analysis, in ComputationSchedulerToModuleScheduler()
529 computation, points_to_analysis, alias_analysis, size_func, in ComputationSchedulerToModuleScheduler()
545 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler() argument
556 *computation, sequence, alias_analysis, size_function, in ListMemoryScheduler()
565 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler() argument
[all …]
Dhlo_memory_scheduler.h66 const HloAliasAnalysis& alias_analysis,
76 const HloAliasAnalysis& alias_analysis,
86 const HloAliasAnalysis& alias_analysis,
99 const HloAliasAnalysis& alias_analysis,
107 const HloAliasAnalysis& alias_analysis,
Dcopy_insertion.cc279 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile() argument
285 if (!IndicesToCopyForWhile(alias_analysis.dataflow_analysis(), xla_while, in AddCopiesForWhile()
337 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation() argument
487 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover() argument
489 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) { in CopyRemover()
494 for (const HloBuffer& buffer : alias_analysis.buffers()) { in CopyRemover()
984 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional() argument
989 if (!IndicesToCopyForConditional(alias_analysis.dataflow_analysis(), in AddCopiesForConditional()
1014 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloAliasAnalysis> alias_analysis, in AddCopiesToResolveInterference()
1021 TF_RETURN_IF_ERROR(AddCopiesForWhile(*alias_analysis, instruction)); in AddCopiesToResolveInterference()
[all …]
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 …]
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()
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 …]
Dhlo_live_range.h44 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis,
101 const HloAliasAnalysis& alias_analysis, in HloLiveRange() argument
104 alias_analysis_(alias_analysis), in HloLiveRange()
Dbuffer_assignment.cc586 if (alias_analysis().BufferLivesOut(buffer)) { in AddAssignment()
600 if (alias_analysis().ValueLivesOut(hlo_value)) { in AddAssignment()
884 alias_analysis().GetBufferContainingValue(*value).values()) { in ToProto()
1029 if (assignment->alias_analysis().BufferLivesOut(hlo_buffer)) { in MaybeAssignBuffer()
1086 if (assignment->alias_analysis().BufferLivesOut(hlo_buffer) && in MaybeAssignBuffer()
1197 !assignment->alias_analysis().BufferLivesOut(*hlo_buffer)) { in AssignSingleHloBuffer()
1254 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations() local
1256 for (const HloBuffer& buffer : alias_analysis.buffers()) { in AssignBuffersForComputations()
1310 sorted_buffers, [&post_order_position, &alias_analysis, assignment]( in AssignBuffersForComputations()
1319 const bool a_live_out = alias_analysis.BufferLivesOut(*a); in AssignBuffersForComputations()
[all …]
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,
Dbuffer_assignment.h461 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis() function
501 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment() argument
507 alias_analysis_(std::move(alias_analysis)), in BufferAssignment()
590 return [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in DefaultColorer()
591 for (HloValue* value : alias_analysis->dataflow_analysis().values()) { in DefaultColorer()
Dmemory_space_assignment.h159 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis() argument
167 alias_analysis_(std::move(alias_analysis)), in MemorySpaceAssignmentCostAnalysis()
796 const HloAliasAnalysis& alias_analysis, const Options& options);
813 const HloAliasAnalysis& alias_analysis);
819 const HloAliasAnalysis& alias_analysis);
950 const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap() argument
955 alias_analysis_(alias_analysis), in AlternateMemoryBestFitHeap()
1000 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis() function
Dcopy_insertion.h87 virtual Status AddCopiesForConditional(const HloAliasAnalysis& alias_analysis,
Dhlo_live_range.cc23 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run() argument
26 new HloLiveRange(schedule, alias_analysis, module_scoped_analysis)); in Run()
Dmemory_space_assignment.cc94 const HloAliasAnalysis& alias_analysis, const HloLiveRange& hlo_live_range, in FindCrossProgramPrefetchCandidate() argument
97 for (const HloBuffer& buffer : alias_analysis.buffers()) { in FindCrossProgramPrefetchCandidate()
138 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
140 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
145 alternate_mem_bandwidth_bytes_per_second, std::move(alias_analysis), in Create()
2649 const HloAliasAnalysis& alias_analysis, in Run() argument
2659 alias_analysis); in Run()
2665 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment() argument
2666 TF_RETURN_IF_ERROR(FindAllocationSequence(hlo_live_range, alias_analysis)); in RunMemorySpaceAssignment()
2691 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.h210 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.cc55 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis() local
140 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace() local
142 HloLiveRange::Run(module->schedule(), *alias_analysis, in AssignMemorySpace()
147 MemorySpaceAssignment::Run(module, *hlo_live_range, *alias_analysis, in AssignMemorySpace()
241 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); local
242 HloBuffer& buffer = alias_analysis->GetUniqueBufferAt(instruction, index);
322 TF_ASSIGN_OR_RETURN(auto alias_analysis, HloAliasAnalysis::Run(&module)); in Create()
324 HloLiveRange::Run(module.schedule(), *alias_analysis, in Create()
330 std::move(alias_analysis), std::move(hlo_live_range), in Create()
385 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis() argument
[all …]
Dbuffer_assignment_test.cc585 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F() argument
588 id < alias_analysis->dataflow_analysis().values().size(); id++) { in TEST_F()
589 auto& value = alias_analysis->dataflow_analysis().GetValue(id); in TEST_F()
651 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F() argument
653 id < alias_analysis->dataflow_analysis().values().size(); id++) { in TEST_F()
654 auto& value = alias_analysis->dataflow_analysis().GetValue(id); in TEST_F()
655 auto& buffer = alias_analysis->GetBufferContainingValue(value); 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.cc71 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult() argument
72 : alias_analysis_(alias_analysis) {} in PerFunctionResult()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
DBUILD108 "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
DBUILD476 "//tensorflow/compiler/xla/service/llvm_ir:alias_analysis",

12