Home
last modified time | relevance | path

Searched refs:cuda_executor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_rng.cc236 gpu::GpuExecutor* cuda_executor = in initialize_curand() local
238 if (cuda_executor == nullptr) { in initialize_curand()
245 gpu::GpuRng* rng = new gpu::GpuRng(cuda_executor); in initialize_curand()
Dcuda_fft.cc570 gpu::GpuExecutor *cuda_executor = in initialize_cufft() local
572 if (cuda_executor == nullptr) { in initialize_cufft()
578 return new gpu::CUDAFft(cuda_executor); in initialize_cufft()
Dcuda_dnn.cc4656 gpu::GpuExecutor* cuda_executor = in initialize_cudnn() local
4658 if (cuda_executor == nullptr) { in initialize_cudnn()
4664 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn()
Dcuda_blas.cc3762 gpu::GpuExecutor *cuda_executor = in initialize_cublas() local
3764 if (cuda_executor == nullptr) { in initialize_cublas()
3771 gpu::CUDABlas *blas = new gpu::CUDABlas(cuda_executor); in initialize_cublas()