Searched refs:MinimumMemoryForModule (Results 1 – 5 of 5) sorted by relevance
115 static StatusOr<int64> MinimumMemoryForModule(
535 TF_ASSIGN_OR_RETURN(*peak_memory, HeapSimulator::MinimumMemoryForModule( in ComputationSchedulerToModuleScheduler()
42 StatusOr<int64> HeapSimulator::MinimumMemoryForModule( in MinimumMemoryForModule() function in xla::HeapSimulator
741 HeapSimulator::MinimumMemoryForModule(schedule, buffer_size_)); in ComputeSummaryStats()
99 HeapSimulator::MinimumMemoryForModule(schedule, size_fn).ValueOrDie()); in TEST_F()