Home
last modified time | relevance | path

Searched defs:alias_analysis (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dside_effect_analysis.h43 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo()
49 Region* region, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in SideEffectAnalysisInfo()
Dside_effect_analysis.cc69 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis) { in FindAccessedResources()
188 const ResourceAliasAnalysis::Info& alias_analysis) { in OpIsDeclaration()
200 Operation* op, const ResourceAliasAnalysis::Info& alias_analysis, in GetResourceIdsByValue()
295 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in AnalyzeFunction()
323 Region* region, const TF::ResourceAliasAnalysis::Info& alias_analysis) { in AnalyzeRegion()
474 ResourceAliasAnalysis alias_analysis(module); in SideEffectAnalysis() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc416 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper()
438 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler()
521 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()
545 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler()
565 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler()
583 const HloAliasAnalysis& alias_analysis, in DefaultMemoryScheduler()
641 const HloAliasAnalysis& alias_analysis, in DefaultModuleScheduler()
Dheap_simulator.cc69 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
84 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
99 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
119 const HloAliasAnalysis& alias_analysis, in Run()
140 const HloAliasAnalysis& alias_analysis, in Run()
158 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
Dhlo_live_range.h101 const HloAliasAnalysis& alias_analysis, in HloLiveRange()
Dcopy_insertion.cc279 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
337 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation()
487 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
984 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional()
Dloop_schedule_linearizer.cc71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
Dhlo_live_range.cc23 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
Dhlo_module_group_metadata.h210 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis() function
Dbuffer_assignment.h461 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis() function
501 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment()
Dmemory_space_assignment.h159 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis()
950 const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap()
1000 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis() function
Dhlo_memory_scheduler_test.cc46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation() local
Dmemory_space_assignment.cc94 const HloAliasAnalysis& alias_analysis, const HloLiveRange& hlo_live_range, in FindCrossProgramPrefetchCandidate()
2649 const HloAliasAnalysis& alias_analysis, in Run()
2665 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment()
2691 const HloAliasAnalysis& alias_analysis) { in FindAllocationSequence()
Dhlo_alias_analysis.cc531 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run() local
Dbuffer_assignment.cc1254 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations() local
1380 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignPresetBuffers() local
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
241 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); local
385 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis()
Dbuffer_assignment_test.cc585 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
651 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
Dheap_simulator_test.cc211 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_device_inference.cc71 FuncOp func_op, const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult()