Searched refs:safe_alloc_frontier (Results 1 – 4 of 4) sorted by relevance
87 uint64 safe_alloc_frontier = dst_device->SafeAllocFrontier(0); in SameWorkerRecvDone() local89 &safe_alloc_frontier]() { in SameWorkerRecvDone()90 safe_alloc_frontier = dst_device->SafeAllocFrontier(safe_alloc_frontier); in SameWorkerRecvDone()91 return safe_alloc_frontier; in SameWorkerRecvDone()93 if (parsed.dst.type == "GPU" && safe_alloc_frontier > 0) { in SameWorkerRecvDone()
130 uint64 safe_alloc_frontier = col_ctx_->device->SafeAllocFrontier(0); in ContinueAfterInputCopy() local132 std::function<uint64()> freed_by_func = [this, &safe_alloc_frontier]() { in ContinueAfterInputCopy()133 safe_alloc_frontier = in ContinueAfterInputCopy()134 col_ctx_->device->SafeAllocFrontier(safe_alloc_frontier); in ContinueAfterInputCopy()135 return safe_alloc_frontier; in ContinueAfterInputCopy()137 if (safe_alloc_frontier > 0) { in ContinueAfterInputCopy()152 (safe_alloc_frontier == 0)); in ContinueAfterInputCopy()
269 uint64 safe_alloc_frontier = dst_device->SafeAllocFrontier(0); in SameWorkerRecvDone() local270 bool sync_dst_compute = (safe_alloc_frontier == 0); in SameWorkerRecvDone()271 std::function<uint64()> freed_by_func = [dst_device, &safe_alloc_frontier]() { in SameWorkerRecvDone()272 safe_alloc_frontier = dst_device->SafeAllocFrontier(safe_alloc_frontier); in SameWorkerRecvDone()273 return safe_alloc_frontier; in SameWorkerRecvDone()
649 uint64 safe_alloc_frontier = 0; in MaybeCopyTensorToGPU() local650 std::function<uint64()> freed_by_func = [this, &safe_alloc_frontier]() { in MaybeCopyTensorToGPU()651 safe_alloc_frontier = SafeAllocFrontier(safe_alloc_frontier); in MaybeCopyTensorToGPU()652 return safe_alloc_frontier; in MaybeCopyTensorToGPU()