Home
last modified time | relevance | path

Searched refs:cudaSetDevice (Results 1 – 8 of 8) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h140 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()
150 cudaError_t err = cudaSetDevice(device_); in initializeDeviceProp()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudart_stub.cc80 cudaError_t CUDART_CB cudaSetDevice(int device) { in cudaSetDevice() function
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc319 CUDA_RETURN_IF_ERROR(cudaSetDevice(devices[i])); in GetCommunicator()
324 CUDA_RETURN_IF_ERROR(cudaSetDevice(saved_device)); in GetCommunicator()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc918 cudaSetDevice(cuda_device_id); in ConvertAfterShapes()
948 cudaSetDevice(old_cuda_device); in ConvertAfterShapes()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1033 err = cudaSetDevice(platform_gpu_id.value()); in CreateDevices()
1062 err = cudaSetDevice(original_device); in CreateDevices()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc698 auto err = cudaSetDevice(platform_gpu_id); in AllocateCalibrationResources()
/external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
Dmpi_ops.cc583 cudaSetDevice(mpi_global.device); in BackgroundThreadLoop()
/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_ops.cc584 cudaSetDevice(mpi_global.device); in BackgroundThreadLoop()