Searched defs:peak_memory (Results 1 – 3 of 3) sorted by relevance
421 const MemorySchedulerPostprocessor& postprocessor, int64* peak_memory) { in ScheduleComputationHelper()443 const MemorySchedulerPostprocessor& postprocessor, int64* peak_memory) { in DFSMemoryScheduler()526 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()555 const MemorySchedulerPostprocessor& postprocessor, int64* peak_memory) { in ListMemoryScheduler()578 const MemorySchedulerPostprocessor& postprocessor, int64* peak_memory) { in PostOrderMemoryScheduler()599 const MemorySchedulerPostprocessor& postprocessor, int64* peak_memory) { in DefaultMemoryScheduler()656 const BufferValue::SizeFunction& size_function, int64* peak_memory) { in DefaultModuleScheduler()713 const ModuleSchedulerAlgorithm& algorithm, int64* peak_memory) { in ScheduleModule()
144 int64_t peak_memory; in TEST_F() local
1811 int64_t peak_memory = tracker.memory_usage(); in ComputePeakMemory() local1873 int64_t peak_memory = memory_tracker.memory_usage(); in RematerializeComputation() local