Home
last modified time | relevance | path

Searched refs:cudaGetDeviceCount (Results 1 – 3 of 3) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceCuda.h61 cudaError_t status = cudaGetDeviceCount(&num_devices); in initializeDeviceProp()
120 cudaError_t err = cudaGetDeviceCount(&num_devices); in initializeDeviceProp()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudart_stub.cc99 cudaError_t CUDART_CB cudaGetDeviceCount(int* count) { in cudaGetDeviceCount() function
/external/tensorflow/tensorflow/tools/test/
Dgpu_info_lib.py140 libcudart.cudaGetDeviceCount(ct.byref(device_count))