Searched refs:memory_by_computation_ (Results 1 – 3 of 3) sorted by relevance
111 memory_by_computation_(memory_by_computation) { in ListScheduler()234 auto it = memory_by_computation_.find(c); in BytesFreedIfScheduled()235 if (it != memory_by_computation_.end()) { in BytesFreedIfScheduled()377 memory_by_computation_; member in xla::__anone86b2dc90111::ListScheduler
282 if (memory_by_computation_ != nullptr) { in RunComputation()284 instruction, alloc_size_by_instruction, *memory_by_computation_); in RunComputation()360 memory_by_computation_(memory_by_computation) { in HeapSimulator()
176 memory_by_computation_; variable