Searched defs:memory_by_computation (Results 1 – 4 of 4) sorted by relevance
84 memory_by_computation) { in Run()109 memory_by_computation) in ListScheduler()408 memory_by_computation, in ScheduleComputationHelper()429 memory_by_computation, in DFSMemoryScheduler()511 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in ComputationSchedulerToModuleScheduler() local536 memory_by_computation, in ListMemoryScheduler()556 memory_by_computation, in PostOrderMemoryScheduler()574 memory_by_computation, in DefaultMemoryScheduler()
70 memory_by_computation) { in MinimumMemoryForComputation()117 memory_by_computation) { in Run()333 memory_by_computation) in HeapSimulator()450 memory_by_computation) { in AccountForSubcomputationMemory()
243 memory_by_computation) {} in AccountForSubcomputationMemory()
206 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in TEST_F() local