Searched refs:PluginConfig (Results 1 – 22 of 22) 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 | 84 PluginConfig plugin_config;
149 int ordinal, const PluginConfig& plugin_config) = 0;
|
| D | stream_executor_pimpl.h | 74 const PluginConfig &plugin_config = PluginConfig());
|
| D | plugin_registry.cc | 211 if (plugin_id == PluginConfig::kDefault) { \
|
| D | stream_executor_internal.h | 382 std::function<StreamExecutorInterface *(const PluginConfig &)>;
|
| D | stream_executor_pimpl.cc | 64 PlatformKind platform_kind, const PluginConfig &plugin_config) { in StreamExecutorImplementationFromPlatformKind()
158 const PluginConfig &plugin_config) in StreamExecutor()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.cc | 47 config.plugin_config = PluginConfig(); in ExecutorForDevice()
53 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | host_gpu_executor.h | 46 explicit HostExecutor(const PluginConfig &plugin_config);
209 const PluginConfig plugin_config_;
|
| D | host_platform.h | 58 int ordinal, const PluginConfig& config) override;
|
| D | host_gpu_executor.cc | 37 HostExecutor::HostExecutor(const PluginConfig &plugin_config) in HostExecutor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.cc | 50 config.plugin_config = PluginConfig(); in ExecutorForDevice()
57 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | executor.h | 54 explicit XlaInterpreterExecutor(const PluginConfig &plugin_config);
212 const PluginConfig plugin_config_;
|
| D | platform.h | 45 int ordinal, const PluginConfig& config) override;
|
| 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 | 70 int ordinal, const PluginConfig& config) override;
|
| D | rocm_gpu_executor.cc | 968 *internal::MakeROCMExecutorImplementation() = [](const PluginConfig& config) { in initialize_rocm_gpu_executor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 149 config.plugin_config = PluginConfig(); in ExecutorForDevice()
155 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | cuda_platform.h | 72 int ordinal, const PluginConfig& config) override;
|
| D | cuda_gpu_executor.cc | 1184 *internal::MakeCUDAExecutorImplementation() = [](const PluginConfig& config) { in initialize_cuda_gpu_executor()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 48 explicit GpuExecutor(const PluginConfig& plugin_config) in GpuExecutor()
339 PluginConfig plugin_config_;
|