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()441 memory_by_computation, in DFSMemoryScheduler()523 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in ComputationSchedulerToModuleScheduler() local548 memory_by_computation, in ListMemoryScheduler()568 memory_by_computation, in PostOrderMemoryScheduler()586 memory_by_computation, in DefaultMemoryScheduler()
72 memory_by_computation) { in MinimumMemoryForComputation()122 memory_by_computation) { in Run()339 memory_by_computation) in HeapSimulator()465 memory_by_computation) { in AccountForSubcomputationMemory()
262 memory_by_computation) {} in AccountForSubcomputationMemory()
207 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in TEST_F() local