| 467 | } |
| 468 | |
| 469 | dnn::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 | |
| 479 | blas::BlasSupport *StreamExecutor::AsBlas() { |
| 480 | absl::MutexLock lock(&mu_); |
no test coverage detected