Searched refs:BlasSupport (Results 1 - 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
H A Dstream.cc1880 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 Dplugin_registry.h53 typedef blas::BlasSupport* (*BlasFactory)(internal::StreamExecutorInterface*);
H A Dstream_executor_internal.h269 // Creates a new BlasSupport object, ownership is transferred to the caller.
274 virtual blas::BlasSupport *CreateBlas() { return nullptr; }
H A Dstream_executor_pimpl.h459 // 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 Dstream_executor_pimpl.cc336 blas::BlasSupport *blas_support = AsBlas();
395 blas::BlasSupport *StreamExecutor::AsBlas() {
H A Dblas.h170 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 Dcuda_blas.h46 // This satisfies the platform-agnostic BlasSupport interface.
54 class CUDABlas : public blas::BlasSupport {
H A Dcuda_gpu_executor.h182 blas::BlasSupport *CreateBlas() override;
H A Dcuda_gpu_executor.cc671 blas::BlasSupport *CUDAExecutor::CreateBlas() {
H A Dcuda_blas.cc2805 *parent) -> gpu::blas::BlasSupport * {
/external/tensorflow/tensorflow/stream_executor/host/
H A Dhost_gpu_executor.cc212 blas::BlasSupport *HostExecutor::CreateBlas() {
H A Dhost_gpu_executor.h175 blas::BlasSupport *CreateBlas() override;

Completed in 542 milliseconds