Searched defs:peak_memory (Results 1 – 3 of 3) sorted by relevance
421 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in ScheduleComputationHelper()443 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in DFSMemoryScheduler()535 int64_t* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()565 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in ListMemoryScheduler()588 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in PostOrderMemoryScheduler()609 const MemorySchedulerPostprocessor& postprocessor, int64_t* peak_memory) { in DefaultMemoryScheduler()668 int64_t* peak_memory) { in DefaultModuleScheduler()727 int64_t* peak_memory) { in ScheduleModule()
144 int64_t peak_memory; in TEST_F() local
1850 int64_t peak_memory = tracker.memory_usage(); in ComputePeakMemory() local1925 int64_t peak_memory = memory_tracker.memory_usage(); in RematerializeComputation() local