Home
last modified time | relevance | path

Searched refs:plugin_id (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.cc73 PluginId plugin_id, const std::string& plugin_name, FACTORY_TYPE factory, in RegisterFactoryInternal() argument
77 if (factories->find(plugin_id) != factories->end()) { in RegisterFactoryInternal()
85 (*factories)[plugin_id] = factory; in RegisterFactoryInternal()
86 plugin_names_[plugin_id] = plugin_name; in RegisterFactoryInternal()
92 PluginId plugin_id, const std::map<PluginId, FACTORY_TYPE>& factories, in GetFactoryInternal() argument
94 auto iter = factories.find(plugin_id); in GetFactoryInternal()
96 iter = generic_factories.find(plugin_id); in GetFactoryInternal()
100 absl::StrFormat("Plugin ID %p not registered.", plugin_id)); in GetFactoryInternal()
109 PluginId plugin_id) { in SetDefaultFactory() argument
110 if (!HasFactory(platform_id, plugin_kind, plugin_id)) { in SetDefaultFactory()
[all …]
Dplugin_registry.h64 port::Status RegisterFactory(Platform::Id platform_id, PluginId plugin_id,
70 port::Status RegisterFactoryForAllPlatforms(PluginId plugin_id,
84 PluginId plugin_id);
95 PluginId plugin_id);
102 PluginId plugin_id);
124 port::Status RegisterFactoryInternal(PluginId plugin_id,
132 PluginId plugin_id, const std::map<PluginId, FactoryT>& factories,
167 Platform::Id platform_id, PluginId plugin_id, const std::string& name, \
171 Platform::Id platform_id, PluginId plugin_id); \
174 PlatformKind platform_kind, PluginId plugin_id)