Searched defs:peak_memory (Results 1 – 3 of 3) sorted by relevance
409 int64* peak_memory) { in ScheduleComputationHelper()430 int64* peak_memory) { in DFSMemoryScheduler()509 int64* peak_memory) -> StatusOr<HloSchedule> { in ComputationSchedulerToModuleScheduler()537 int64* peak_memory) { in ListMemoryScheduler()557 int64* peak_memory) { in PostOrderMemoryScheduler()575 int64* peak_memory) { in DefaultMemoryScheduler()630 const BufferValue::SizeFunction& size_function, int64* peak_memory) { in DefaultModuleScheduler()686 const ModuleSchedulerAlgorithm& algorithm, int64* peak_memory) { in ScheduleModule()
144 int64 peak_memory; in TEST_F() local
1414 int64 peak_memory = tracker.memory_usage(); in ComputePeakMemory() local1472 int64 peak_memory = memory_tracker.memory_usage(); in RematerializeComputation() local