HomeSort by relevance Sort by last modified time
    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_ids[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