Searched defs:memory_by_computation (Results 1 – 4 of 4) sorted by relevance
85 memory_by_computation) { in Run()110 memory_by_computation) in ListScheduler()420 memory_by_computation, in ScheduleComputationHelper()442 memory_by_computation, in DFSMemoryScheduler()537 absl::flat_hash_map<const HloComputation*, int64_t> memory_by_computation; in ComputationSchedulerToModuleScheduler() local564 memory_by_computation, in ListMemoryScheduler()587 memory_by_computation, in PostOrderMemoryScheduler()608 memory_by_computation, in DefaultMemoryScheduler()
77 memory_by_computation) { in MinimumMemoryForComputation()125 memory_by_computation) { in Run()342 memory_by_computation) in HeapSimulator()468 memory_by_computation) { in AccountForSubcomputationMemory()
268 memory_by_computation) {} in AccountForSubcomputationMemory()
206 absl::flat_hash_map<const HloComputation*, int64_t> memory_by_computation; in TEST_F() local