HomeSort by relevance Sort by last modified time
    Searched refs:blas_ (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
plugin.cc 27 : blas_(kDefault), dnn_(kDefault), fft_(kDefault), rng_(kDefault) {}
30 return blas_ == rhs.blas_ && dnn_ == rhs.dnn_ && fft_ == rhs.fft_ &&
35 blas_ = blas;
plugin.h 77 PluginId blas() const { return blas_; }
83 PluginId blas_, dnn_, fft_, rng_; member in class:perftools::gputools::PluginConfig
stream_executor_pimpl.cc 397 if (blas_ != nullptr) {
398 return blas_.get();
401 blas_.reset(implementation_->CreateBlas());
402 return blas_.get();
stream_executor_pimpl.h 604 std::unique_ptr<blas::BlasSupport> blas_ GUARDED_BY(mu_);
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_blas.h 159 cublasHandle_t blas_ GUARDED_BY(mu_);
cuda_blas.cc 441 cublasStatus_t ret = wrap::cublasCreate(parent_, &blas_);
451 : parent_(CHECK_NOTNULL(parent)), blas_(nullptr) {}
454 if (blas_ != nullptr) {
455 wrap::cublasDestroy(parent_, blas_);
462 CHECK(blas_ != nullptr);
464 wrap::cublasSetStream(parent_, blas_, AsCUDAStreamValue(stream));
615 CHECK(blas_ != nullptr);
620 ScopedCublasPointerMode pointer_mode{parent_, blas_};
626 ScopedCublasMathMode math_mode{parent_, blas_};
633 cublasStatus_t ret = cublas_func(parent_, blas_, args...)
    [all...]

Completed in 406 milliseconds