Searched refs:memory_by_computation (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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, peak_memory); in ScheduleComputationHelper()
429 size_function, memory_by_computation, in ScheduleComputationHelper()
441 memory_by_computation, in DFSMemoryScheduler() argument
510 &memory_by_computation)); in DFSMemoryScheduler()
523 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in ComputationSchedulerToModuleScheduler() local
[all …]
|
| D | hlo_memory_scheduler.h | 69 memory_by_computation,
79 memory_by_computation,
89 memory_by_computation,
102 memory_by_computation,
|
| D | heap_simulator.h | 126 memory_by_computation = nullptr);
162 memory_by_computation = nullptr);
182 memory_by_computation = nullptr);
262 memory_by_computation) {} in AccountForSubcomputationMemory() argument
300 memory_by_computation) override;
|
| D | heap_simulator.cc | 72 memory_by_computation) { in MinimumMemoryForComputation() argument
78 memory_by_computation)); in MinimumMemoryForComputation()
122 memory_by_computation) { in Run() argument
124 /*schedule=*/nullptr, memory_by_computation); in Run()
339 memory_by_computation) in HeapSimulator() argument
346 memory_by_computation_(memory_by_computation) { in HeapSimulator()
465 memory_by_computation) { in AccountForSubcomputationMemory() argument
470 auto it = memory_by_computation.find(c); in AccountForSubcomputationMemory()
471 if (it != memory_by_computation.end()) { in AccountForSubcomputationMemory()
|
| D | heap_simulator_test.cc | 207 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in TEST_F() local
208 memory_by_computation[cond_computation] = 5; in TEST_F()
209 memory_by_computation[body_computation] = 16; in TEST_F()
218 *alias_analysis, size_fn, &memory_by_computation) in TEST_F()
|