Searched refs:device_contexts_ (Results 1 – 2 of 2) sorted by relevance
337 for (auto ctx : device_contexts_) ctx->Unref(); in ~BaseGPUDevice()390 device_contexts_.push_back(new GPUDeviceContext( in Init()396 gpu_device_info_->default_context = device_contexts_[0]; in Init()478 auto ctx = device_contexts_[mapped_stream]; in FillContextMap()516 GPUDeviceContext* gpu_device_context = device_contexts_[0]; in ComputeHelper()603 GPUDeviceContext* gpu_device_context = device_contexts_[0]; in ConsumeListOfAccessedTensors()618 GPUDeviceContext* gpu_device_context = device_contexts_[0]; in ComputeAsync()678 device_contexts_[0]->CopyCPUTensorToDevice(&from, this, copy, in MaybeCopyTensorToGPU()
149 std::vector<GPUDeviceContext*> device_contexts_; variable