Searched defs:peak_memory (Results 1 – 3 of 3) sorted by relevance
421 int64* peak_memory) { in ScheduleComputationHelper()442 int64* peak_memory) { in DFSMemoryScheduler()521 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()549 int64* peak_memory) { in ListMemoryScheduler()569 int64* peak_memory) { in PostOrderMemoryScheduler()587 int64* peak_memory) { in DefaultMemoryScheduler()642 const BufferValue::SizeFunction& size_function, int64* peak_memory) { in DefaultModuleScheduler()698 const ModuleSchedulerAlgorithm& algorithm, int64* peak_memory) { in ScheduleModule()
144 int64 peak_memory; in TEST_F() local
1799 int64 peak_memory = tracker.memory_usage(); in ComputePeakMemory() local1861 int64 peak_memory = memory_tracker.memory_usage(); in RematerializeComputation() local