Home
last modified time | relevance | path

Searched refs:BlasSupport (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1882 bool (blas::BlasSupport::*blas_func)(Stream *, Args...), in operator ()()
1890 bool (blas::BlasSupport::*blas_func)(Stream *, Args...),
1896 Stream *stream, bool (blas::BlasSupport::*blas_func)(Stream *, Args...), in Run()
1900 if (blas::BlasSupport *blas = stream->parent_->AsBlas()) { in Run()
1921 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1931 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1942 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1953 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1965 return impl(this, &blas::BlasSupport::DoBlasAxpy, elem_count, alpha, x, incx, in ThenBlasAxpy()
1977 return impl(this, &blas::BlasSupport::DoBlasAxpy, elem_count, alpha, x, incx, in ThenBlasAxpy()
[all …]
Dplugin_registry.h53 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*);
Dstream_executor_pimpl.h515 blas::BlasSupport *AsBlas();
658 std::unique_ptr<blas::BlasSupport> blas_ GUARDED_BY(mu_);
Dstream_executor_internal.h318 virtual blas::BlasSupport *CreateBlas() { return nullptr; } in CreateBlas()
Dstream_executor_pimpl.cc373 blas::BlasSupport *blas_support = AsBlas(); in GetBlasGemmAlgorithms()
449 blas::BlasSupport *StreamExecutor::AsBlas() { in AsBlas()
Dblas.h173 class BlasSupport {
175 virtual ~BlasSupport() {} in ~BlasSupport()
1386 BlasSupport() {} in BlasSupport() function
1389 SE_DISALLOW_COPY_AND_ASSIGN(BlasSupport);
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.h50 class ROCMBlas : public blas::BlasSupport {
Drocm_gpu_executor.cc655 blas::BlasSupport* GpuExecutor::CreateBlas() { in CreateBlas()
Drocm_blas.cc2341 -> blas::BlasSupport * { in initialize_rocblas()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.h53 class CUDABlas : public blas::BlasSupport {
Dcuda_gpu_executor.cc774 blas::BlasSupport* GpuExecutor::CreateBlas() { in CreateBlas()
Dcuda_blas.cc2803 [](internal::StreamExecutorInterface *parent) -> blas::BlasSupport * { in initialize_cublas()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h175 blas::BlasSupport *CreateBlas() override;
Dhost_gpu_executor.cc211 blas::BlasSupport *HostExecutor::CreateBlas() { in CreateBlas()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h210 blas::BlasSupport* CreateBlas() override;