HomeSort by relevance Sort by last modified time
    Searched defs:PluginConfig (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
plugin.cc 24 PLUGIN_REGISTRY_DEFINE_PLUGIN_ID(PluginConfig::kDefault);
26 PluginConfig::PluginConfig()
29 bool PluginConfig::operator==(const PluginConfig& rhs) const {
34 PluginConfig& PluginConfig::SetBlas(PluginId blas) {
39 PluginConfig& PluginConfig::SetDnn(PluginId dnn) {
44 PluginConfig& PluginConfig::SetFft(PluginId fft)
    [all...]
plugin.h 46 // A PluginConfig describes the set of plugins to be used by a StreamExecutor
50 // A PluginConfig may be passed to the StreamExecutor constructor - the plugins
61 class PluginConfig {
67 PluginConfig();
69 bool operator==(const PluginConfig& rhs) const;
72 PluginConfig& SetBlas(PluginId blas);
73 PluginConfig& SetDnn(PluginId dnn);
74 PluginConfig& SetFft(PluginId fft);
75 PluginConfig& SetRng(PluginId rng);

Completed in 51 milliseconds