Searched refs:plugin_config (Results 1 – 13 of 13) sorted by relevance
| /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() argument
56 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
70 this, MakeUnique<HostExecutor>(config.plugin_config)); in GetUncachedExecutor()
|
| D | host_gpu_executor.cc | 37 HostExecutor::HostExecutor(const PluginConfig &plugin_config) in HostExecutor() argument
38 : 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 | 50 config.plugin_config = PluginConfig(); in ExecutorForDevice()
57 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig() argument
60 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
75 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 | 54 explicit XlaInterpreterExecutor(const PluginConfig &plugin_config);
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | executor_cache.cc | 46 if (iter.first.plugin_config == config.plugin_config && in GetOrCreate()
86 if (iter.first.plugin_config == config.plugin_config && in Get()
|
| D | platform.h | 84 PluginConfig plugin_config; member
149 int ordinal, const PluginConfig& plugin_config) = 0;
|
| D | stream_executor_pimpl.cc | 64 PlatformKind platform_kind, const PluginConfig &plugin_config) { in StreamExecutorImplementationFromPlatformKind() argument
91 return factory(plugin_config); in StreamExecutorImplementationFromPlatformKind()
158 const PluginConfig &plugin_config) in StreamExecutor() argument
161 platform_kind, plugin_config)), in StreamExecutor()
|
| D | stream_executor_pimpl.h | 74 const PluginConfig &plugin_config = PluginConfig());
|
| /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, MakeUnique<GpuExecutor>(config.plugin_config)); in GetUncachedExecutor()
|
| /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() argument
158 config.plugin_config = plugin_config; in ExecutorForDeviceWithPluginConfig()
172 this, MakeUnique<GpuExecutor>(config.plugin_config)); in GetUncachedExecutor()
|
| /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()
|