Searched refs:SafeAllocFrontier (Results 1 – 6 of 6) sorted by relevance
87 uint64 safe_alloc_frontier = dst_device->SafeAllocFrontier(0); in SameWorkerRecvDone()90 safe_alloc_frontier = dst_device->SafeAllocFrontier(safe_alloc_frontier); in SameWorkerRecvDone()
130 uint64 safe_alloc_frontier = col_ctx_->device->SafeAllocFrontier(0); in ContinueAfterInputCopy()134 col_ctx_->device->SafeAllocFrontier(safe_alloc_frontier); in ContinueAfterInputCopy()
269 uint64 safe_alloc_frontier = dst_device->SafeAllocFrontier(0); in SameWorkerRecvDone()272 safe_alloc_frontier = dst_device->SafeAllocFrontier(safe_alloc_frontier); in SameWorkerRecvDone()
251 virtual uint64 SafeAllocFrontier(uint64 old_value) { return 0; } in SafeAllocFrontier() function
111 uint64 SafeAllocFrontier(uint64 old_value) override;
651 safe_alloc_frontier = SafeAllocFrontier(safe_alloc_frontier); in MaybeCopyTensorToGPU()1914 uint64 BaseGPUDevice::SafeAllocFrontier(uint64 old_value) { in SafeAllocFrontier() function in tensorflow::BaseGPUDevice