Searched refs:plugin_kind (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 30 std::string PluginKindString(PluginKind plugin_kind) { in PluginKindString() argument
31 switch (plugin_kind) { in PluginKindString()
108 PluginKind plugin_kind, in SetDefaultFactory() argument
110 if (!HasFactory(platform_id, plugin_kind, plugin_id)) { in SetDefaultFactory()
121 << ", PluginKind: " << PluginKindString(plugin_kind) in SetDefaultFactory()
126 switch (plugin_kind) { in SetDefaultFactory()
141 << static_cast<int>(plugin_kind); in SetDefaultFactory()
149 PluginKind plugin_kind, in HasFactory() argument
151 switch (plugin_kind) { in HasFactory()
162 << PluginKindString(plugin_kind); in HasFactory()
[all …]
|
| D | plugin_registry.h | 83 bool SetDefaultFactory(Platform::Id platform_id, PluginKind plugin_kind,
88 bool HasFactory(Platform::Id platform_id, PluginKind plugin_kind,
138 bool HasFactory(const PluginFactories& factories, PluginKind plugin_kind,
|