Searched refs:PluginRegistry (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 43 PluginRegistry::DefaultFactories::DefaultFactories() : in DefaultFactories()
51 /* static */ PluginRegistry* PluginRegistry::instance_ = nullptr;
53 PluginRegistry::PluginRegistry() {} in PluginRegistry() function in stream_executor::PluginRegistry
55 /* static */ PluginRegistry* PluginRegistry::Instance() { in Instance()
58 instance_ = new PluginRegistry(); in Instance()
63 void PluginRegistry::MapPlatformKindToId(PlatformKind platform_kind, in MapPlatformKindToId()
69 port::Status PluginRegistry::RegisterFactoryInternal( in RegisterFactoryInternal()
88 port::StatusOr<FACTORY_TYPE> PluginRegistry::GetFactoryInternal( in GetFactoryInternal()
104 bool PluginRegistry::SetDefaultFactory(Platform::Id platform_id, in SetDefaultFactory()
145 bool PluginRegistry::HasFactory(const PluginFactories& factories, in HasFactory()
[all …]
|
| D | plugin_registry.h | 51 class PluginRegistry {
59 static PluginRegistry* Instance();
121 PluginRegistry();
143 static PluginRegistry* instance_;
161 SE_DISALLOW_COPY_AND_ASSIGN(PluginRegistry);
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 205 return PluginRegistry::Instance() in SupportsBlas()
206 ->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in SupportsBlas()
212 PluginRegistry *registry = PluginRegistry::Instance(); in CreateBlas()
213 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
214 registry->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in CreateBlas()
226 return PluginRegistry::Instance() in SupportsFft()
227 ->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in SupportsFft()
233 PluginRegistry *registry = PluginRegistry::Instance(); in CreateFft()
234 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
235 registry->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in CreateFft()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 656 PluginRegistry* registry = PluginRegistry::Instance(); in CreateBlas()
657 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
658 registry->GetFactory<PluginRegistry::BlasFactory>(rocm::kROCmPlatformId, in CreateBlas()
670 PluginRegistry* registry = PluginRegistry::Instance(); in CreateDnn()
671 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
672 registry->GetFactory<PluginRegistry::DnnFactory>(rocm::kROCmPlatformId, in CreateDnn()
684 PluginRegistry* registry = PluginRegistry::Instance(); in CreateFft()
685 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
686 registry->GetFactory<PluginRegistry::FftFactory>(rocm::kROCmPlatformId, in CreateFft()
698 PluginRegistry* registry = PluginRegistry::Instance(); in CreateRng()
[all …]
|
| D | rocm_rng.cc | 286 auto rocRandAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_rocrand()
291 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::RngFactory>( in initialize_rocrand()
317 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocrand()
|
| D | rocm_fft.cc | 586 auto rocFftAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_rocfft()
591 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_rocfft()
610 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocfft()
|
| D | rocm_blas.cc | 2332 auto rocBlasAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_rocblas()
2337 PluginRegistry::Instance() in initialize_rocblas()
2338 ->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_rocblas()
2366 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocblas()
|
| D | rocm_dnn.cc | 4483 auto miopenAlreadyRegistered = PluginRegistry::Instance()->HasFactory( in initialize_miopen()
4488 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_miopen()
4514 PluginRegistry::Instance()->SetDefaultFactory( in initialize_miopen()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 775 PluginRegistry *registry = PluginRegistry::Instance(); in CreateBlas()
776 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
777 registry->GetFactory<PluginRegistry::BlasFactory>(cuda::kCudaPlatformId, in CreateBlas()
789 PluginRegistry *registry = PluginRegistry::Instance(); in CreateDnn()
790 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
791 registry->GetFactory<PluginRegistry::DnnFactory>(cuda::kCudaPlatformId, in CreateDnn()
803 PluginRegistry *registry = PluginRegistry::Instance(); in CreateFft()
804 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
805 registry->GetFactory<PluginRegistry::FftFactory>(cuda::kCudaPlatformId, in CreateFft()
817 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 | 538 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_cufft()
556 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cufft()
|
| D | cuda_dnn.cc | 4120 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_cudnn()
4145 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cudnn()
|
| D | cuda_blas.cc | 2801 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_cublas()
2827 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
|