Searched refs:memory_by_computation (Results 1 – 6 of 6) sorted by relevance
82 memory_by_computation) { in Run() argument84 memory_by_computation); in Run()107 memory_by_computation) in ListScheduler() argument111 memory_by_computation_(memory_by_computation) { in ListScheduler()407 memory_by_computation) { in ScheduleComputationHelper() argument411 memory_by_computation); in ScheduleComputationHelper()414 memory_by_computation); in ScheduleComputationHelper()424 memory_by_computation) { in DFSMemoryScheduler() argument495 memory_by_computation) { in ListMemoryScheduler() argument497 memory_by_computation); in ListMemoryScheduler()[all …]
50 memory_by_computation);58 memory_by_computation);66 memory_by_computation);76 memory_by_computation);
104 memory_by_computation = nullptr);134 memory_by_computation = nullptr);144 memory_by_computation = nullptr);235 memory_by_computation) {} in AccountForSubcomputationMemory() argument259 memory_by_computation) override;
62 memory_by_computation) { in MinimumMemoryForComputation() argument68 memory_by_computation)); in MinimumMemoryForComputation()94 memory_by_computation) { in Run() argument96 /*schedule=*/nullptr, memory_by_computation); in Run()354 memory_by_computation) in HeapSimulator() argument360 memory_by_computation_(memory_by_computation) { in HeapSimulator()531 memory_by_computation) { in AccountForSubcomputationMemory() argument536 auto it = memory_by_computation.find(c); in AccountForSubcomputationMemory()537 if (it != memory_by_computation.end()) { in AccountForSubcomputationMemory()
295 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in TEST_F() local296 memory_by_computation[cond_computation] = 17; in TEST_F()297 memory_by_computation[body_computation] = 16; in TEST_F()310 *points_to_analysis, size_fn, &memory_by_computation) in TEST_F()
206 absl::flat_hash_map<const HloComputation*, int64> memory_by_computation; in TEST_F() local207 memory_by_computation[cond_computation] = 5; in TEST_F()208 memory_by_computation[body_computation] = 16; in TEST_F()216 *points_to_analysis, size_fn, &memory_by_computation) in TEST_F()