| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler.cc | 404 const HloAliasAnalysis& alias_analysis, in ScheduleComputationHelper()
426 const HloAliasAnalysis& alias_analysis, in DFSMemoryScheduler()
509 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()
533 const HloAliasAnalysis& alias_analysis, in ListMemoryScheduler()
553 const HloAliasAnalysis& alias_analysis, in PostOrderMemoryScheduler()
571 const HloAliasAnalysis& alias_analysis, in DefaultMemoryScheduler()
629 const HloAliasAnalysis& alias_analysis, in DefaultModuleScheduler()
|
| D | heap_simulator.cc | 67 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
81 const HloAliasAnalysis& alias_analysis, in MinimumMemoryForComputation()
95 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
114 const HloAliasAnalysis& alias_analysis, in Run()
134 const HloAliasAnalysis& alias_analysis, in Run()
152 const HloAliasAnalysis& alias_analysis, HloLiveRange* hlo_live_range) { in RunComputation()
|
| D | hlo_live_range.h | 101 const HloAliasAnalysis& alias_analysis, in HloLiveRange()
|
| D | hlo_live_range.cc | 23 const HloSchedule& schedule, const HloAliasAnalysis& alias_analysis, in Run()
|
| D | hlo_module_group_metadata.h | 210 HloAliasAnalysis* alias_analysis(HloModule* module) const { in alias_analysis() function
|
| D | buffer_assignment.h | 448 HloAliasAnalysis& alias_analysis() const { return *alias_analysis_; } in alias_analysis() function
487 std::unique_ptr<HloAliasAnalysis> alias_analysis, in BufferAssignment()
|
| D | copy_insertion.cc | 256 Status AddCopiesForWhile(const HloAliasAnalysis& alias_analysis, in AddCopiesForWhile()
316 Status AddCopiesForConditional(const HloAliasAnalysis& alias_analysis, in AddCopiesForConditional()
474 CopyRemover(const HloModule& module, const HloAliasAnalysis& alias_analysis, in CopyRemover()
|
| D | hlo_memory_scheduler_test.cc | 46 std::unique_ptr<HloAliasAnalysis> alias_analysis = in PeakMemoryUseOfEntryComputation() local
|
| D | hlo_alias_analysis.cc | 497 auto alias_analysis = absl::WrapUnique(new HloAliasAnalysis(module)); in Run() local
|
| D | memory_space_assignment.h | 628 const HloAliasAnalysis& alias_analysis, in AlternateMemoryBestFitHeap()
|
| D | buffer_assignment.cc | 1239 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignBuffersForComputations() local
1369 const HloAliasAnalysis& alias_analysis = assignment->alias_analysis(); in AssignPresetBuffers() local
|
| D | heap_simulator_test.cc | 210 std::unique_ptr<HloAliasAnalysis> alias_analysis = in TEST_F() local
|
| D | memory_space_assignment.cc | 1173 const HloAliasAnalysis& alias_analysis, in Run()
|
| D | buffer_assignment_test.cc | 585 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
651 auto colorer = [](HloAliasAnalysis* alias_analysis, const HloOrdering&) { in TEST_F()
|
| D | memory_space_assignment_test.cc | 55 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpaceUsingCostAnalysis() local
118 auto alias_analysis = HloAliasAnalysis::Run(module).ValueOrDie(); in AssignMemorySpace() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | side_effect_analysis.cc | 270 Operation* op, const ResourceAliasAnalysis& alias_analysis) { in FindAccessedResources()
315 const ResourceAliasAnalysis& alias_analysis) { in OpIsDeclaration()
401 FuncOp func_op, const ResourceAliasAnalysis& alias_analysis) { in AnalyzeFunction()
429 Region* region, const ResourceAliasAnalysis& alias_analysis) { in AnalyzeRegion()
545 ResourceAliasAnalysis alias_analysis(op); in SideEffectAnalysis() local
|