Searched refs:plugin_kind (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin_registry.cc | 27 string PluginKindString(PluginKind plugin_kind) { in PluginKindString() argument
28 switch (plugin_kind) { in PluginKindString()
105 PluginKind plugin_kind, in SetDefaultFactory() argument
107 if (!HasFactory(platform_id, plugin_kind, plugin_id)) { in SetDefaultFactory()
118 << ", PluginKind: " << PluginKindString(plugin_kind) in SetDefaultFactory()
123 switch (plugin_kind) { in SetDefaultFactory()
138 << static_cast<int>(plugin_kind); in SetDefaultFactory()
146 PluginKind plugin_kind, in HasFactory() argument
148 switch (plugin_kind) { in HasFactory()
159 << PluginKindString(plugin_kind); in HasFactory()
[all …]
|
| D | plugin_registry.h | 84 bool SetDefaultFactory(Platform::Id platform_id, PluginKind plugin_kind,
89 bool HasFactory(Platform::Id platform_id, PluginKind plugin_kind,
139 bool HasFactory(const PluginFactories& factories, PluginKind plugin_kind,
|