Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h53 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*); typedef
108 std::map<PluginId, BlasFactory> blas;
Dplugin_registry.cc242 EMIT_PLUGIN_SPECIALIZATIONS(BlasFactory, blas, "BLAS");
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc206 ->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in SupportsBlas()
213 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
214 registry->GetFactory<PluginRegistry::BlasFactory>(kHostPlatformId, in CreateBlas()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc657 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
658 registry->GetFactory<PluginRegistry::BlasFactory>(rocm::kROCmPlatformId, in CreateBlas()
Drocm_blas.cc2338 ->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_rocblas()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc776 port::StatusOr<PluginRegistry::BlasFactory> status = in CreateBlas()
777 registry->GetFactory<PluginRegistry::BlasFactory>(cuda::kCudaPlatformId, in CreateBlas()
Dcuda_blas.cc2801 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::BlasFactory>( in initialize_cublas()