| 146 | } |
| 147 | |
| 148 | bool PluginRegistry::HasFactory(const PluginFactories& factories, |
| 149 | PluginKind plugin_kind, |
| 150 | PluginId plugin_id) const { |
| 151 | switch (plugin_kind) { |
| 152 | case PluginKind::kBlas: |
| 153 | return factories.blas.find(plugin_id) != factories.blas.end(); |
| 154 | case PluginKind::kDnn: |
| 155 | return factories.dnn.find(plugin_id) != factories.dnn.end(); |
| 156 | case PluginKind::kFft: |
| 157 | return factories.fft.find(plugin_id) != factories.fft.end(); |
| 158 | case PluginKind::kRng: |
| 159 | return factories.rng.find(plugin_id) != factories.rng.end(); |
| 160 | default: |
| 161 | LOG(ERROR) << "Invalid plugin kind specified: " |
| 162 | << PluginKindString(plugin_kind); |
| 163 | return false; |
| 164 | } |
| 165 | } |
| 166 | |
| 167 | bool PluginRegistry::HasFactory(Platform::Id platform_id, |
| 168 | PluginKind plugin_kind, |
no test coverage detected