Searched refs:tf_gpu_id_ (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_device.cc | 310 tf_gpu_id_(tf_gpu_id), in BaseGPUDevice() 354 tf_gpu_id_.value()); in InitScratchBuffers() 364 tf_gpu_id_.value()); in InitScratchBuffers() 373 auto executor_status = GpuIdUtil::ExecutorForTfGpuId(tf_gpu_id_); in Init() 376 tf_gpu_id_.value()); in Init() 389 tf_gpu_id_, i, executor_, options.config.gpu_options())); in Init() 400 GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in Init() 429 options.env, strings::StrCat("gpu_private_", tf_gpu_id_.value()), in Init() 510 " on GPU ", tf_gpu_id_.value(), " stream[", stream_id, in ComputeOpKernelDebugString() 627 << op_kernel->type_string() << " on GPU" << tf_gpu_id_ << " stream[" in ComputeAsync() [all …]
|
D | gpu_device.h | 104 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in gpu_id() 152 TfGpuId tf_gpu_id_; variable
|