Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h103 cudaGetDevice(&device_); in initializeDeviceProp()
117 cudaGetDevice(&device_); in initializeDeviceProp()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudart_stub.cc58 cudaError_t CUDART_CB cudaGetDevice(int* device) { in cudaGetDevice() function
/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h246 cudaGetDevice(&dev); in GetCuda3DLaunchConfig()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.cc312 CUDA_RETURN_IF_ERROR(cudaGetDevice(&saved_device)); in GetCommunicator()
/external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
Dmpi_ops.cc806 cudaGetDevice(&mpi_global.device); in InitializeMPIOnce()
/external/tensorflow/tensorflow/contrib/mpi_collectives/
Dmpi_ops.cc807 cudaGetDevice(&mpi_global.device); in InitializeMPIOnce()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc889 auto err = cudaGetDevice(&old_cuda_device); in ConvertAfterShapes()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1016 cudaError_t err = cudaGetDevice(&original_device); in CreateDevices()