Searched refs:PluginKind (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | plugin_registry.cc | 30 std::string PluginKindString(PluginKind plugin_kind) { in PluginKindString()
32 case PluginKind::kBlas: in PluginKindString()
34 case PluginKind::kDnn: in PluginKindString()
36 case PluginKind::kFft: in PluginKindString()
38 case PluginKind::kRng: in PluginKindString()
40 case PluginKind::kInvalid: in PluginKindString()
108 PluginKind plugin_kind, in SetDefaultFactory()
127 case PluginKind::kBlas: in SetDefaultFactory()
130 case PluginKind::kDnn: in SetDefaultFactory()
133 case PluginKind::kFft: in SetDefaultFactory()
[all …]
|
| D | plugin_registry.h | 83 bool SetDefaultFactory(Platform::Id platform_id, PluginKind plugin_kind,
88 bool HasFactory(Platform::Id platform_id, PluginKind plugin_kind,
138 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 | 286 rocm::kROCmPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_rocrand()
317 rocm::kROCmPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_rocrand()
|
| D | rocm_fft.cc | 619 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
643 rocm::kROCmPlatformId, PluginKind::kFft, gpu::kRocFftPlugin); in initialize_rocfft()
|
| D | rocm_blas.cc | 1170 rocm::kROCmPlatformId, PluginKind::kBlas, gpu::kRocBlasPlugin); in initialize_rocblas()
1204 rocm::kROCmPlatformId, PluginKind::kBlas, gpu::kRocBlasPlugin); in initialize_rocblas()
|
| D | rocm_dnn.cc | 5019 rocm::kROCmPlatformId, PluginKind::kDnn, gpu::kMIOpenPlugin); in initialize_miopen()
5050 rocm::kROCmPlatformId, PluginKind::kDnn, gpu::kMIOpenPlugin); in initialize_miopen()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_rng.cc | 260 cuda::kCudaPlatformId, PluginKind::kRng, gpu::kGpuRandPlugin); in initialize_curand()
|
| D | cuda_fft.cc | 616 cuda::kCudaPlatformId, PluginKind::kFft, gpu::kCuFftPlugin); in initialize_cufft()
|
| D | cuda_blas.cc | 1700 cuda::kCudaPlatformId, PluginKind::kBlas, kCuBlasPlugin); in DoBlasInternalImpl()
|
| D | cuda_dnn.cc | 6590 cuda::kCudaPlatformId, PluginKind::kDnn, gpu::kCuDnnPlugin); in initialize_cudnn()
|