MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / AsDnn

Method AsDnn

tensorflow/stream_executor/stream_executor_pimpl.cc:469–477  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

467}
468
469dnn::DnnSupport *StreamExecutor::AsDnn() {
470 absl::MutexLock lock(&mu_);
471 if (dnn_ != nullptr) {
472 return dnn_.get();
473 }
474
475 dnn_.reset(implementation_->CreateDnn());
476 return dnn_.get();
477}
478
479blas::BlasSupport *StreamExecutor::AsBlas() {
480 absl::MutexLock lock(&mu_);

Callers 8

GetCudnnVersionFunction · 0.80
stream.ccFile · 0.80
GetCudnnVersionFunction · 0.80
PrintPlatformInfoFunction · 0.80
GetCudnnVersionFunction · 0.80

Calls 3

getMethod · 0.45
resetMethod · 0.45
CreateDnnMethod · 0.45

Tested by

no test coverage detected