Home
last modified time | relevance | path

Searched refs:device_contexts_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc337 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()
Dgpu_device.h149 std::vector<GPUDeviceContext*> device_contexts_; variable