Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc310 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 …]
Dgpu_device.h104 TF_CHECK_OK(GpuIdManager::TfToPlatformGpuId(tf_gpu_id_, &platform_gpu_id)); in gpu_id()
152 TfGpuId tf_gpu_id_; variable