Searched refs:BlasSupport (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 1434 bool (blas::BlasSupport::*blas_func)(Stream *, Args...), in operator ()()
1442 bool (blas::BlasSupport::*blas_func)(Stream *, Args...),
1448 Stream *stream, bool (blas::BlasSupport::*blas_func)(Stream *, Args...), in Run()
1452 if (blas::BlasSupport *blas = stream->parent_->AsBlas()) { in Run()
1473 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1484 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1496 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1508 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, in ThenBlasAsum()
1521 return impl(this, &blas::BlasSupport::DoBlasAxpy, elem_count, alpha, x, incx, in ThenBlasAxpy()
1534 return impl(this, &blas::BlasSupport::DoBlasAxpy, elem_count, alpha, x, incx, in ThenBlasAxpy()
[all …]
|
| D | plugin_registry.h | 52 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*);
|
| D | stream_executor_pimpl.cc | 333 blas::BlasSupport *blas_support = AsBlas(); in GetBlasGemmAlgorithms()
343 blas::BlasSupport *blas_support = AsBlas(); in CreateBlasLtMatmulPlan()
355 blas::BlasSupport *blas_support = AsBlas(); in GetBlasLtMatmulAlgorithms()
434 blas::BlasSupport *StreamExecutor::AsBlas() { in AsBlas()
|
| D | stream_executor_pimpl.h | 507 blas::BlasSupport *AsBlas();
684 std::unique_ptr<blas::BlasSupport> blas_ TF_GUARDED_BY(mu_);
|
| D | stream_executor_internal.h | 321 virtual blas::BlasSupport *CreateBlas() { return nullptr; } in CreateBlas()
|
| D | blas.h | 242 class BlasSupport {
244 virtual ~BlasSupport() {} in ~BlasSupport()
1522 BlasSupport() {} in BlasSupport() function
1525 SE_DISALLOW_COPY_AND_ASSIGN(BlasSupport);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 56 class CUDABlas : public blas::BlasSupport {
|
| D | cuda_gpu_executor.cc | 746 blas::BlasSupport* GpuExecutor::CreateBlas() { in CreateBlas()
|
| D | cuda_blas.cc | 3761 [](internal::StreamExecutorInterface *parent) -> blas::BlasSupport * { in initialize_cublas()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 73 class ROCMBlas : public blas::BlasSupport {
|
| D | rocm_gpu_executor.cc | 654 blas::BlasSupport* GpuExecutor::CreateBlas() { in CreateBlas()
|
| D | rocm_blas.cc | 2694 -> blas::BlasSupport * { in initialize_rocblas()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 152 blas::BlasSupport *CreateBlas() override;
|
| D | host_gpu_executor.cc | 303 blas::BlasSupport *HostExecutor::CreateBlas() { in CreateBlas()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 208 blas::BlasSupport* CreateBlas() override;
|