Searched refs:memory_by_computation (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_memory_scheduler.cc | 85 memory_by_computation) { in Run() argument
87 memory_by_computation); in Run()
110 memory_by_computation) in ListScheduler() argument
114 memory_by_computation_(memory_by_computation) { in ListScheduler()
420 memory_by_computation, in ScheduleComputationHelper() argument
426 size_function, memory_by_computation, postprocessor, in ScheduleComputationHelper()
430 size_function, memory_by_computation, in ScheduleComputationHelper()
442 memory_by_computation, in DFSMemoryScheduler() argument
521 &memory_by_computation)); in DFSMemoryScheduler()
537 absl::flat_hash_map<const HloComputation*, int64_t> memory_by_computation; in ComputationSchedulerToModuleScheduler() local
[all …]
|
| H A D | hlo_memory_scheduler.h | 77 memory_by_computation,
87 memory_by_computation,
97 memory_by_computation,
110 memory_by_computation,
|
| H A D | heap_simulator.h | 132 memory_by_computation = nullptr);
168 memory_by_computation = nullptr);
188 memory_by_computation = nullptr);
268 memory_by_computation) {} in AccountForSubcomputationMemory() argument
306 memory_by_computation) override;
|
| H A D | heap_simulator.cc | 77 memory_by_computation) { in MinimumMemoryForComputation() argument
82 HeapSimulator::Options(), memory_by_computation)); in MinimumMemoryForComputation()
125 memory_by_computation) { in Run() argument
127 /*schedule=*/nullptr, memory_by_computation); in Run()
342 memory_by_computation) in HeapSimulator() argument
349 memory_by_computation_(memory_by_computation) { in HeapSimulator()
468 memory_by_computation) { in AccountForSubcomputationMemory() argument
473 auto it = memory_by_computation.find(c); in AccountForSubcomputationMemory()
474 if (it != memory_by_computation.end()) { in AccountForSubcomputationMemory()
|
| H A D | heap_simulator_test.cc | 206 absl::flat_hash_map<const HloComputation*, int64_t> memory_by_computation; in TEST_F() local
207 memory_by_computation[cond_computation] = 5; in TEST_F()
208 memory_by_computation[body_computation] = 16; in TEST_F()
217 *alias_analysis, size_fn, &memory_by_computation) in TEST_F()
|