Searched refs:peak_memory (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_memory_scheduler.cc | 421 int64* peak_memory) { in ScheduleComputationHelper() argument
426 size_function, memory_by_computation, peak_memory); in ScheduleComputationHelper()
430 peak_memory); in ScheduleComputationHelper()
442 int64* peak_memory) { in DFSMemoryScheduler() argument
506 if (peak_memory) { in DFSMemoryScheduler()
508 *peak_memory, HeapSimulator::MinimumMemoryForComputation( in DFSMemoryScheduler()
521 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler() argument
534 if (peak_memory) { in ComputationSchedulerToModuleScheduler()
535 TF_ASSIGN_OR_RETURN(*peak_memory, HeapSimulator::MinimumMemoryForModule( in ComputationSchedulerToModuleScheduler()
549 int64* peak_memory) { in ListMemoryScheduler() argument
[all …]
|
| D | hlo_memory_scheduler.h | 70 int64* peak_memory);
80 int64* peak_memory);
90 int64* peak_memory);
103 int64* peak_memory);
108 const LogicalBuffer::SizeFunction& size_function, int64* peak_memory);
117 int64* peak_memory = nullptr);
|
| D | hlo_memory_scheduler_test.cc | 144 int64 peak_memory; in TEST_F() local
149 &peak_memory)); in TEST_F()
172 peak_memory); in TEST_F()
|
| D | hlo_rematerialization.cc | 1799 int64 peak_memory = tracker.memory_usage(); in ComputePeakMemory() local
1806 peak_memory = in ComputePeakMemory()
1807 std::max<int64>(peak_memory, tracker.memory_usage() + callee_usage); in ComputePeakMemory()
1811 << HumanReadableNumBytes(peak_memory); in ComputePeakMemory()
1812 return peak_memory; in ComputePeakMemory()
1861 int64 peak_memory = memory_tracker.memory_usage(); in RematerializeComputation() local
1966 peak_memory = std::max<int64>(peak_memory, in RematerializeComputation()
1968 VLOG(3) << "peak memory usage = " << HumanReadableNumBytes(peak_memory); in RematerializeComputation()
1981 VLOG(1) << " peak memory usage now " << HumanReadableNumBytes(peak_memory) in RematerializeComputation()
1987 computation_peak_memory_.at(computation) = peak_memory; in RematerializeComputation()
|
| /external/llvm-project/clang/utils/analyzer/ |
| D | SATestBuild.py | 433 peak_memory = 0
477 peak_memory = max(peak_memory, mem)
485 return execution_time, peak_memory
513 peak_memory = 0
541 peak_memory = max(peak_memory, mem)
553 return execution_time, peak_memory
|