Searched defs:memory_by_computation (Results 1 – 5 of 5) sorted by relevance
82 memory_by_computation) { in Run()107 memory_by_computation) in ListScheduler()407 memory_by_computation) { in ScheduleComputationHelper()424 memory_by_computation) { in DFSMemoryScheduler()495 memory_by_computation) { in ListMemoryScheduler()505 memory_by_computation) { in PostOrderMemoryScheduler()514 memory_by_computation) { in DefaultMemoryScheduler()576 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in ScheduleModule() local
62 memory_by_computation) { in MinimumMemoryForComputation()94 memory_by_computation) { in Run()354 memory_by_computation) in HeapSimulator()531 memory_by_computation) { in AccountForSubcomputationMemory()
295 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in TEST_F() local
235 memory_by_computation) {} in AccountForSubcomputationMemory()
206 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in TEST_F() local