Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_rng.cc294 gpu::GpuExecutor* rocm_executor = in initialize_rocrand() local
296 if (rocm_executor == nullptr) { in initialize_rocrand()
303 gpu::GpuRng* rng = new gpu::GpuRng(rocm_executor); in initialize_rocrand()
Drocm_fft.cc594 gpu::GpuExecutor *rocm_executor = in initialize_rocfft() local
596 if (rocm_executor == nullptr) { in initialize_rocfft()
603 return new gpu::ROCMFft(rocm_executor); in initialize_rocfft()
Drocm_blas.cc2342 gpu::GpuExecutor *rocm_executor = in initialize_rocblas() local
2344 if (rocm_executor == nullptr) { in initialize_rocblas()
2352 gpu::ROCMBlas *blas = new gpu::ROCMBlas(rocm_executor); in initialize_rocblas()
Drocm_dnn.cc4491 gpu::GpuExecutor* rocm_executor = in initialize_miopen() local
4493 if (rocm_executor == nullptr) { in initialize_miopen()
4500 gpu::MIOpenSupport* dnn = new gpu::MIOpenSupport(rocm_executor); in initialize_miopen()