Searched refs:MinimumMemoryForModule (Results 1 – 5 of 5) sorted by relevance
121 static StatusOr<int64_t> MinimumMemoryForModule(
551 TF_ASSIGN_OR_RETURN(*peak_memory, HeapSimulator::MinimumMemoryForModule( in ComputationSchedulerToModuleScheduler()
48 StatusOr<int64_t> HeapSimulator::MinimumMemoryForModule( in MinimumMemoryForModule() function in xla::HeapSimulator
721 HeapSimulator::MinimumMemoryForModule(schedule, buffer_size_)); in ComputeSummaryStats()
98 HeapSimulator::MinimumMemoryForModule(schedule, size_fn).ValueOrDie()); in TEST_F()