Searched refs:PluginRegistry (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 46 PluginRegistry::DefaultFactories::DefaultFactories() : in DefaultFactories()
54 /* static */ PluginRegistry* PluginRegistry::instance_ = nullptr;
56 PluginRegistry::PluginRegistry() {} in PluginRegistry() function in stream_executor::PluginRegistry
58 /* static */ PluginRegistry* PluginRegistry::Instance() { in Instance()
61 instance_ = new PluginRegistry(); in Instance()
66 void PluginRegistry::MapPlatformKindToId(PlatformKind platform_kind, in MapPlatformKindToId()
72 port::Status PluginRegistry::RegisterFactoryInternal( in RegisterFactoryInternal()
91 port::StatusOr<FACTORY_TYPE> PluginRegistry::GetFactoryInternal( in GetFactoryInternal()
107 bool PluginRegistry::SetDefaultFactory(Platform::Id platform_id, in SetDefaultFactory()
148 bool PluginRegistry::HasFactory(const PluginFactories& factories, in HasFactory()
[all …]
|
| D | plugin_registry.h | 50 class PluginRegistry {
58 static PluginRegistry* Instance();
120 PluginRegistry();
142 static PluginRegistry* instance_;
160 SE_DISALLOW_COPY_AND_ASSIGN(PluginRegistry);
166 port::Status PluginRegistry::RegisterFactory<PluginRegistry::FACTORY_TYPE>( \
168 PluginRegistry::FACTORY_TYPE factory); \
170 port::StatusOr<PluginRegistry::FACTORY_TYPE> PluginRegistry::GetFactory( \
173 port::StatusOr<PluginRegistry::FACTORY_TYPE> PluginRegistry::GetFactory( \
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 297 return PluginRegistry::Instance() in SupportsBlas()
298 ->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in SupportsBlas()
304 PluginRegistry *registry = PluginRegistry::Instance(); in CreateBlas()
305 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
306 registry->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in CreateBlas()
318 return PluginRegistry::Instance() in SupportsFft()
319 ->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in SupportsFft()
325 PluginRegistry *registry = PluginRegistry::Instance(); in CreateFft()
326 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
327 registry->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in CreateFft()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 655 PluginRegistry* registry = PluginRegistry::Instance(); in CreateBlas()
656 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
657 registry->GetFactory<PluginRegistry::BlasFactory>(rocm::kROCmPlatformId, in CreateBlas()
669 PluginRegistry* registry = PluginRegistry::Instance(); in CreateDnn()
670 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
671 registry->GetFactory<PluginRegistry::DnnFactory>(rocm::kROCmPlatformId, in CreateDnn()
683 PluginRegistry* registry = PluginRegistry::Instance(); in CreateFft()
684 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
685 registry->GetFactory<PluginRegistry::FftFactory>(rocm::kROCmPlatformId, in CreateFft()
697 PluginRegistry* registry = PluginRegistry::Instance(); in CreateRng()
[all …]
|
| D | rocm_rng.cc | 285 auto rocRandAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_rocrand()
290 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::RngFactory>( in initialize_rocrand()
316 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocrand()
|
| D | rocm_fft.cc | 591 auto rocFftAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_rocfft()
596 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_rocfft()
615 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocfft()
|
| D | rocm_blas.cc | 2685 auto rocBlasAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_rocblas()
2690 PluginRegistry::Instance() in initialize_rocblas()
2691 ->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_rocblas()
2719 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocblas()
|
| D | rocm_dnn.cc | 5079 auto miopenAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_miopen()
5084 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_miopen()
5110 PluginRegistry::Instance()->SetDefaultFactory( in initialize_miopen()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 747 PluginRegistry* registry = PluginRegistry::Instance(); in CreateBlas()
748 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
749 registry->GetFactory<PluginRegistry::BlasFactory>(cuda::kCudaPlatformId, in CreateBlas()
761 PluginRegistry* registry = PluginRegistry::Instance(); in CreateDnn()
762 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
763 registry->GetFactory<PluginRegistry::DnnFactory>(cuda::kCudaPlatformId, in CreateDnn()
775 PluginRegistry* registry = PluginRegistry::Instance(); in CreateFft()
776 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
777 registry->GetFactory<PluginRegistry::FftFactory>(cuda::kCudaPlatformId, in CreateFft()
789 PluginRegistry* registry = PluginRegistry::Instance(); in CreateRng()
[all …]
|
| D | cuda_rng.cc | 233 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::RngFactory>( in initialize_curand()
259 PluginRegistry::Instance()->SetDefaultFactory( in initialize_curand()
|
| D | cuda_fft.cc | 567 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_cufft()
585 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cufft()
|
| D | cuda_dnn.cc | 4653 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_cudnn()
4678 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cudnn()
|
| D | cuda_blas.cc | 3759 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_cublas()
3785 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cublas()
|
| /external/mockito/src/main/java/org/mockito/internal/configuration/plugins/ |
| D | Plugins.java | 18 private static final PluginRegistry registry = new PluginRegistry();
|
| D | PluginRegistry.java | 15 class PluginRegistry { class
31 PluginRegistry() { in PluginRegistry() method in PluginRegistry
|