Home
last modified time | relevance | path

Searched refs:BlasFactory (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h52 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*); typedef
107 std::map<PluginId, BlasFactory> blas;
176 DECLARE_PLUGIN_SPECIALIZATIONS(BlasFactory);
Dplugin_registry.cc246 EMIT_PLUGIN_SPECIALIZATIONS(BlasFactory, blas, "BLAS");
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc298 ->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in SupportsBlas()
305 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
306 registry->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in CreateBlas()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc656 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
657 registry->GetFactory<PluginRegistry::BlasFactory>(rocm::kROCmPlatformId, in CreateBlas()
Drocm_blas.cc2691 ->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_rocblas()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc748 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
749 registry->GetFactory<PluginRegistry::BlasFactory>(cuda::kCudaPlatformId, in CreateBlas()
Dcuda_blas.cc3759 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_cublas()