Searched refs:PluginConfig (Results 1 – 21 of 21) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | plugin.cc | 23 PLUGIN_REGISTRY_DEFINE_PLUGIN_ID(PluginConfig::kDefault);
25 PluginConfig::PluginConfig() in PluginConfig() function in stream_executor::PluginConfig
28 bool PluginConfig::operator==(const PluginConfig& rhs) const { in operator ==()
33 PluginConfig& PluginConfig::SetBlas(PluginId blas) { in SetBlas()
38 PluginConfig& PluginConfig::SetDnn(PluginId dnn) { in SetDnn()
43 PluginConfig& PluginConfig::SetFft(PluginId fft) { in SetFft()
48 PluginConfig& PluginConfig::SetRng(PluginId rng) { in SetRng()
|
| D | plugin.h | 60 class PluginConfig {
66 PluginConfig();
68 bool operator==(const PluginConfig& rhs) const;
71 PluginConfig& SetBlas(PluginId blas);
72 PluginConfig& SetDnn(PluginId dnn);
73 PluginConfig& SetFft(PluginId fft);
74 PluginConfig& SetRng(PluginId rng);
|
| D | platform.h | 86 PluginConfig plugin_config;
160 int ordinal, const PluginConfig& plugin_config) = 0;
|
| D | plugin_registry.cc | 215 if (plugin_id == PluginConfig::kDefault) { \
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.cc | 52 config.plugin_config = PluginConfig(); in ExecutorForDevice()
58 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | host_gpu_executor.h | 46 explicit HostExecutor(const PluginConfig &plugin_config);
180 const PluginConfig plugin_config_;
|
| D | host_platform.h | 60 int ordinal, const PluginConfig& config) override;
|
| D | host_gpu_executor.cc | 43 HostExecutor::HostExecutor(const PluginConfig &plugin_config) in HostExecutor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.cc | 54 config.plugin_config = PluginConfig(); in ExecutorForDevice()
61 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | platform.h | 49 int ordinal, const PluginConfig& config) override;
|
| D | executor.h | 53 explicit XlaInterpreterExecutor(const PluginConfig &plugin_config);
210 const PluginConfig plugin_config_;
|
| D | executor.cc | 31 const PluginConfig &plugin_config) in XlaInterpreterExecutor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 115 config.plugin_config = PluginConfig(); in ExecutorForDevice()
121 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | rocm_platform.h | 73 int ordinal, const PluginConfig& config) override;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 152 config.plugin_config = PluginConfig(); in ExecutorForDevice()
158 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | cuda_platform.h | 73 int ordinal, const PluginConfig& config) override;
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 48 explicit GpuExecutor(const PluginConfig& plugin_config) in GpuExecutor()
338 PluginConfig plugin_config_;
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_platform.h | 95 const ::stream_executor::PluginConfig& plugin_config) override { in ExecutorForDeviceWithPluginConfig()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_internal.h | 67 int ordinal, const PluginConfig& plugin_config) override;
|
| D | stream_executor.cc | 712 int ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function_test.cc | 320 int ordinal, const se::PluginConfig& config) override { in TEST()
|