Searched refs:plugin_config (Results 1 – 14 of 14) sorted by relevance
| /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() argument
61 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
75 this, absl::make_unique<HostExecutor>(config.plugin_config), in GetUncachedExecutor()
|
| D | host_gpu_executor.cc | 43 HostExecutor::HostExecutor(const PluginConfig &plugin_config) in HostExecutor() argument
44 : plugin_config_(plugin_config) {} in HostExecutor()
|
| D | host_gpu_executor.h | 46 explicit HostExecutor(const PluginConfig &plugin_config);
|
| /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() argument
64 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
79 this, absl::make_unique<XlaInterpreterExecutor>(config.plugin_config), in GetUncachedExecutor()
|
| D | executor.cc | 31 const PluginConfig &plugin_config) in XlaInterpreterExecutor() argument
32 : plugin_config_(plugin_config) {} in XlaInterpreterExecutor()
|
| D | executor.h | 53 explicit XlaInterpreterExecutor(const PluginConfig &plugin_config);
|
| /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() argument
124 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
138 this, absl::make_unique<GpuExecutor>(config.plugin_config), in GetUncachedExecutor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | executor_cache.cc | 47 if (iter.first.plugin_config == config.plugin_config && in GetOrCreate()
89 if (iter.first.plugin_config == config.plugin_config && in Get()
|
| D | platform.h | 86 PluginConfig plugin_config; member
160 int ordinal, const PluginConfig& plugin_config) = 0;
|
| /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() argument
161 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
175 this, absl::make_unique<GpuExecutor>(config.plugin_config), in GetUncachedExecutor()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_platform.h | 95 const ::stream_executor::PluginConfig& plugin_config) override { in ExecutorForDeviceWithPluginConfig()
98 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 48 explicit GpuExecutor(const PluginConfig& plugin_config) in GpuExecutor() argument
55 plugin_config_(plugin_config) {} in GpuExecutor()
|
| /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() argument
715 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
|