OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:cuda_gpu_id
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_id_utils.h
34
gpu::Platform* gpu_manager, CudaGpuId
cuda_gpu_id
) {
35
return gpu_manager->ExecutorForDevice(
cuda_gpu_id
.value());
38
CudaGpuId
cuda_gpu_id
) {
39
return ExecutorForCudaGpuId(GPUMachineManager(),
cuda_gpu_id
);
46
// Verify that the
cuda_gpu_id
associated with a TfGpuId is legitimate.
48
const CudaGpuId
cuda_gpu_id
= GpuIdManager::TfToCudaGpuId(tf_gpu_id);
local
50
CHECK_LT(
cuda_gpu_id
.value(), visible_device_count)
51
<< "
cuda_gpu_id
is outside discovered device range."
52
<< " TF GPU id: " << tf_gpu_id << " CUDA GPU id: " <<
cuda_gpu_id
process_state.cc
128
const CudaGpuId
cuda_gpu_id
= GpuIdManager::TfToCudaGpuId(tf_gpu_id);
local
130
new GPUBFCAllocator(
cuda_gpu_id
, total_bytes, options,
136
gpu_allocator = new GPUDebugAllocator(gpu_allocator,
cuda_gpu_id
);
137
gpu_allocator = new GPUNanResetAllocator(gpu_allocator,
cuda_gpu_id
);
142
gpu_allocator = new GPUcudaMallocAllocator(gpu_allocator,
cuda_gpu_id
);
159
md.dev_index =
cuda_gpu_id
.value();
gpu_device.cc
107
const int
cuda_gpu_id
= GpuIdManager::TfToCudaGpuId(tf_gpu_id).value();
local
108
device_prop_ = &Eigen::m_deviceProperties[
cuda_gpu_id
];
682
int32
cuda_gpu_id
;
local
683
if (!strings::safe_strto32(cuda_gpu_id_str, &
cuda_gpu_id
)) {
688
if (
cuda_gpu_id
< 0 ||
cuda_gpu_id
>= gpu_manager->VisibleDeviceCount()) {
690
"'visible_device_list' listed an invalid GPU id '",
cuda_gpu_id
,
694
visible_gpu_order->push_back(CudaGpuId(
cuda_gpu_id
));
780
// '
cuda_gpu_id
', when that virtual device is the only virtual device being
783
CudaGpuId
cuda_gpu_id
,
946
const CudaGpuId
cuda_gpu_id
= valid_
cuda_gpu_id
s[i];
local
1014
CudaGpuId
cuda_gpu_id
= GpuIdManager::TfToCudaGpuId(tf_gpu_id);
local
1086
CudaGpuId
cuda_gpu_id
= GpuIdManager::TfToCudaGpuId(tf_gpu_id);
local
1300
const CudaGpuId
cuda_gpu_id
= visible_gpu_order[i];
local
[
all
...]
Completed in 77 milliseconds