Searched refs:BlasSupport (Results 1 - 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
H A D | stream.cc | 1880 bool (blas::BlasSupport::*blas_func)(Stream *, Args...), 1888 bool (blas::BlasSupport::*blas_func)(Stream *, Args...), 1894 Stream *stream, bool (blas::BlasSupport::*blas_func)(Stream *, Args...), 1898 if (blas::BlasSupport *blas = stream->parent_->AsBlas()) { 1919 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, 1929 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, 1940 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, 1951 return impl(this, &blas::BlasSupport::DoBlasAsum, elem_count, x, incx, 1963 return impl(this, &blas::BlasSupport::DoBlasAxpy, elem_count, alpha, x, incx, 1975 return impl(this, &blas::BlasSupport [all...] |
H A D | plugin_registry.h | 53 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*);
|
H A D | stream_executor_internal.h | 269 // Creates a new BlasSupport object, ownership is transferred to the caller. 274 virtual blas::BlasSupport *CreateBlas() { return nullptr; }
|
H A D | stream_executor_pimpl.h | 459 // Gets-or-creates (creates with memoization) a BlasSupport datatype that can 470 blas::BlasSupport *AsBlas(); 604 std::unique_ptr<blas::BlasSupport> blas_ GUARDED_BY(mu_);
|
H A D | stream_executor_pimpl.cc | 336 blas::BlasSupport *blas_support = AsBlas(); 395 blas::BlasSupport *StreamExecutor::AsBlas() {
|
H A D | blas.h | 170 class BlasSupport { class in namespace:perftools::gputools::blas 172 virtual ~BlasSupport() {} 1336 BlasSupport() {} function in class:perftools::gputools::blas::BlasSupport 1339 SE_DISALLOW_COPY_AND_ASSIGN(BlasSupport); 1343 // BlasSupport base class.
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
H A D | cuda_blas.h | 46 // This satisfies the platform-agnostic BlasSupport interface. 54 class CUDABlas : public blas::BlasSupport {
|
H A D | cuda_gpu_executor.h | 182 blas::BlasSupport *CreateBlas() override;
|
H A D | cuda_gpu_executor.cc | 671 blas::BlasSupport *CUDAExecutor::CreateBlas() {
|
H A D | cuda_blas.cc | 2805 *parent) -> gpu::blas::BlasSupport * {
|
/external/tensorflow/tensorflow/stream_executor/host/ |
H A D | host_gpu_executor.cc | 212 blas::BlasSupport *HostExecutor::CreateBlas() {
|
H A D | host_gpu_executor.h | 175 blas::BlasSupport *CreateBlas() override;
|
Completed in 542 milliseconds