Searched refs:BlasSupport (Results 1 – 15 of 15) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | stream.cc | 1882 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 …]
|
D | plugin_registry.h | 53 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*);
|
D | stream_executor_pimpl.h | 515 blas::BlasSupport *AsBlas(); 658 std::unique_ptr<blas::BlasSupport> blas_ GUARDED_BY(mu_);
|
D | stream_executor_internal.h | 318 virtual blas::BlasSupport *CreateBlas() { return nullptr; } in CreateBlas()
|
D | stream_executor_pimpl.cc | 373 blas::BlasSupport *blas_support = AsBlas(); in GetBlasGemmAlgorithms() 449 blas::BlasSupport *StreamExecutor::AsBlas() { in AsBlas()
|
D | blas.h | 173 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/ |
D | rocm_blas.h | 50 class ROCMBlas : public blas::BlasSupport {
|
D | rocm_gpu_executor.cc | 655 blas::BlasSupport* GpuExecutor::CreateBlas() { in CreateBlas()
|
D | rocm_blas.cc | 2341 -> blas::BlasSupport * { in initialize_rocblas()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_blas.h | 53 class CUDABlas : public blas::BlasSupport {
|
D | cuda_gpu_executor.cc | 774 blas::BlasSupport* GpuExecutor::CreateBlas() { in CreateBlas()
|
D | cuda_blas.cc | 2803 [](internal::StreamExecutorInterface *parent) -> blas::BlasSupport * { in initialize_cublas()
|
/external/tensorflow/tensorflow/stream_executor/host/ |
D | host_gpu_executor.h | 175 blas::BlasSupport *CreateBlas() override;
|
D | host_gpu_executor.cc | 211 blas::BlasSupport *HostExecutor::CreateBlas() { in CreateBlas()
|
/external/tensorflow/tensorflow/stream_executor/gpu/ |
D | gpu_executor.h | 210 blas::BlasSupport* CreateBlas() override;
|