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.cc541 gpu::GpuExecutor *cuda_executor = in initialize_cufft() local
543 if (cuda_executor == nullptr) { in initialize_cufft()
549 return new gpu::CUDAFft(cuda_executor); in initialize_cufft()
Dcuda_dnn.cc4123 gpu::GpuExecutor* cuda_executor = in initialize_cudnn() local
4125 if (cuda_executor == nullptr) { in initialize_cudnn()
4131 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn()
Dcuda_blas.cc2804 gpu::GpuExecutor *cuda_executor = in initialize_cublas() local
2806 if (cuda_executor == nullptr) { in initialize_cublas()
2813 gpu::CUDABlas *blas = new gpu::CUDABlas(cuda_executor); in initialize_cublas()