| 861 | } |
| 862 | |
| 863 | dnn::DnnSupport* GpuExecutor::CreateDnn() { |
| 864 | PluginRegistry *registry = PluginRegistry::Instance(); |
| 865 | port::StatusOr<PluginRegistry::DnnFactory> status = |
| 866 | registry->GetFactory<PluginRegistry::DnnFactory>(cuda::kCudaPlatformId, |
| 867 | plugin_config_.dnn()); |
| 868 | if (!status.ok()) { |
| 869 | LOG(ERROR) << "Unable to retrieve DNN factory: " |
| 870 | << status.status().error_message(); |
| 871 | return nullptr; |
| 872 | } |
| 873 | |
| 874 | return status.ValueOrDie()(this); |
| 875 | } |
| 876 | |
| 877 | fft::FftSupport* GpuExecutor::CreateFft() { |
| 878 | PluginRegistry *registry = PluginRegistry::Instance(); |