Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.cc32 case PluginKind::kBlas: in PluginKindString()
127 case PluginKind::kBlas: in SetDefaultFactory()
152 case PluginKind::kBlas: in HasFactory()
Dplugin.h39 kBlas, enumerator
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2686 rocm::kROCmPlatformId, PluginKind::kBlas, gpu::kRocBlasPlugin); in initialize_rocblas()
2720 rocm::kROCmPlatformId, PluginKind::kBlas, gpu::kRocBlasPlugin); in initialize_rocblas()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3786 cuda::kCudaPlatformId, PluginKind::kBlas, gpu::kCuBlasPlugin); in initialize_cublas()