| 705 | } |
| 706 | |
| 707 | dnn::DnnSupport* GpuExecutor::CreateDnn() { |
| 708 | PluginRegistry* registry = PluginRegistry::Instance(); |
| 709 | port::StatusOr<PluginRegistry::DnnFactory> status = |
| 710 | registry->GetFactory<PluginRegistry::DnnFactory>(rocm::kROCmPlatformId, |
| 711 | plugin_config_.dnn()); |
| 712 | if (!status.ok()) { |
| 713 | LOG(ERROR) << "Unable to retrieve DNN factory: " |
| 714 | << status.status().error_message(); |
| 715 | return nullptr; |
| 716 | } |
| 717 | |
| 718 | return status.ValueOrDie()(this); |
| 719 | } |
| 720 | |
| 721 | fft::FftSupport* GpuExecutor::CreateFft() { |
| 722 | PluginRegistry* registry = PluginRegistry::Instance(); |