Searched refs:PluginKind (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 27 string PluginKindString(PluginKind plugin_kind) { in PluginKindString()
29 case PluginKind::kBlas: in PluginKindString()
31 case PluginKind::kDnn: in PluginKindString()
33 case PluginKind::kFft: in PluginKindString()
35 case PluginKind::kRng: in PluginKindString()
37 case PluginKind::kInvalid: in PluginKindString()
105 PluginKind plugin_kind, in SetDefaultFactory()
124 case PluginKind::kBlas: in SetDefaultFactory()
127 case PluginKind::kDnn: in SetDefaultFactory()
130 case PluginKind::kFft: in SetDefaultFactory()
[all …]
|
| D | plugin_registry.h | 84 bool SetDefaultFactory(Platform::Id platform_id, PluginKind plugin_kind,
89 bool HasFactory(Platform::Id platform_id, PluginKind plugin_kind,
139 bool HasFactory(const PluginFactories& factories, PluginKind plugin_kind,
|
| D | plugin.h | 37 enum class PluginKind { enum
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_rng.cc | 287 rocm::kROCmPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_rocrand()
318 rocm::kROCmPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_rocrand()
|
| D | rocm_fft.cc | 587 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
611 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
|
| D | rocm_blas.cc | 2333 rocm::kROCmPlatformId, PluginKind::kBlas, gpu::kRocBlasPlugin); in initialize_rocblas()
2367 rocm::kROCmPlatformId, PluginKind::kBlas, gpu::kRocBlasPlugin); in initialize_rocblas()
|
| D | rocm_dnn.cc | 4484 rocm::kROCmPlatformId, PluginKind::kDnn, gpu::kMIOpenPlugin); in initialize_miopen()
4515 rocm::kROCmPlatformId, PluginKind::kDnn, gpu::kMIOpenPlugin); in initialize_miopen()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_rng.cc | 260 cuda::kCudaPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_curand()
|
| D | cuda_fft.cc | 557 cuda::kCudaPlatformId, PluginKind::kFft, gpu::kCuFftPlugin); in initialize_cufft()
|
| D | cuda_dnn.cc | 4146 cuda::kCudaPlatformId, PluginKind::kDnn, gpu::kCuDnnPlugin); in initialize_cudnn()
|
| D | cuda_blas.cc | 2828 cuda::kCudaPlatformId, PluginKind::kBlas, gpu::kCuBlasPlugin); in initialize_cublas()
|