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.h62 const TF::ResourceAliasAnalysis::Info& alias_analysis) in SideEffectAnalysisInfo()
71 const TF::ResourceAliasAnalysis::Info& alias_analysis) in SideEffectAnalysisInfo()
Dside_effect_analysis.cc70 const ResourceAliasAnalysis::Info& alias_analysis) { in GetResourceUniqueIdsOrUnknown()
350 const TF::ResourceAliasAnalysis::Info& alias_analysis) { in CollectSideEffectsByResourceId()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler.cc416 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper()
439 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler()
535 int64_t* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()
561 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler()
584 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler()
605 const HloAliasAnalysis& alias_analysis, in DefaultMemoryScheduler()
665 const HloAliasAnalysis& alias_analysis, in DefaultModuleScheduler()
Dheap_simulator.cc74 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
88 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
102 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
122 const HloAliasAnalysis& alias_analysis, in Run()
143 const HloAliasAnalysis& alias_analysis, in Run()
161 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
Dhlo_live_range.h110 const HloAliasAnalysis& alias_analysis, in HloLiveRange()
Dloop_schedule_linearizer.cc71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
Dhlo_module_group_metadata.h209 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis() function
Dbuffer_assignment.h461 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis() function
504 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment()
Dhlo_live_range.cc34 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
Dhlo_memory_scheduler_test.cc46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation() local
Dcopy_insertion.cc288 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
346 Status AddCopiesForInPlaceOperation(const HloAliasAnalysis& alias_analysis, in AddCopiesForInPlaceOperation()
1146 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
1784 const HloAliasAnalysis& alias_analysis, HloInstruction* conditional) { in AddCopiesForConditional()
Dhlo_alias_analysis.cc390 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run() local
Dmemory_space_assignment.cc136 FindCrossProgramPrefetchCandidate(const HloAliasAnalysis& alias_analysis, in FindCrossProgramPrefetchCandidate()
974 const Options& options, const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap()
3450 const HloAliasAnalysis& alias_analysis, in Run()
3466 const HloAliasAnalysis& alias_analysis) { in RunMemorySpaceAssignment()
3499 const HloAliasAnalysis& alias_analysis) { in FindAllocationSequence()
Dmemory_space_assignment.h212 std::unique_ptr<HloAliasAnalysis> alias_analysis, in MemorySpaceAssignmentCostAnalysis()
1261 const HloAliasAnalysis& alias_analysis() { return alias_analysis_; } in alias_analysis() function
Dbuffer_assignment.cc1310 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations() local
1436 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignPresetBuffers() local
Dmemory_space_assignment_test.cc67 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis() local
163 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace() local
273 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); local
417 std::unique_ptr<HloAliasAnalysis> alias_analysis, in FakeMemorySpaceAssignmentCostAnalysis()
Dbuffer_assignment_test.cc603 auto colorer = [&](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
670 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
Dheap_simulator_test.cc210 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_device_inference.cc73 const TF::ResourceAliasAnalysis::Info& alias_analysis) in PerFunctionResult()