OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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