| /external/tensorflow/tensorflow/stream_executor/ |
| D | executor_cache.cc | 47 iter.first.device_options == config.device_options) { in GetOrCreate()
87 iter.first.device_options == config.device_options) { in Get()
|
| D | BUILD | 62 "device_options.h",
117 "device_options.h",
354 name = "device_options",
355 hdrs = ["device_options.h"],
366 "device_options.h",
498 ":device_options",
575 "device_options.h",
|
| D | platform.cc | 84 : ordinal(-1), device_options(DeviceOptions::Default()) {} in StreamExecutorConfig()
87 : ordinal(ordinal_in), device_options(DeviceOptions::Default()) {} in StreamExecutorConfig()
|
| D | platform.h | 87 DeviceOptions device_options; member
|
| D | stream_executor_internal.h | 180 DeviceOptions device_options) = 0;
|
| D | stream_executor_pimpl.cc | 225 DeviceOptions device_options) { in Init() argument
227 return implementation_->Init(device_ordinal, std::move(device_options)); in Init()
|
| D | stream_executor_pimpl.h | 83 port::Status Init(int device_ordinal, DeviceOptions device_options);
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.cc | 48 config.device_options = DeviceOptions::Default(); in ExecutorForDevice()
57 config.device_options = DeviceOptions::Default(); in ExecutorForDeviceWithPluginConfig()
71 auto init_status = executor->Init(config.ordinal, config.device_options); in GetUncachedExecutor()
|
| D | host_gpu_executor.h | 49 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.cc | 51 config.device_options = DeviceOptions::Default(); in ExecutorForDevice()
61 config.device_options = DeviceOptions::Default(); in ExecutorForDeviceWithPluginConfig()
76 auto init_status = executor->Init(config.ordinal, config.device_options); in GetUncachedExecutor()
|
| D | executor.h | 57 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 116 config.device_options = DeviceOptions::Default(); in ExecutorForDevice()
125 config.device_options = DeviceOptions::Default(); in ExecutorForDeviceWithPluginConfig()
139 auto init_status = executor->Init(config.ordinal, config.device_options); in GetUncachedExecutor()
|
| D | BUILD | 56 "//tensorflow/stream_executor:device_options",
|
| D | rocm_gpu_executor.cc | 144 DeviceOptions device_options) { in Init() argument
157 status = GpuDriver::CreateContext(device_ordinal_, device_, device_options, in Init()
|
| D | rocm_driver.cc | 355 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags() argument
363 int device_ordinal, hipDevice_t device, const DeviceOptions& device_options, in CreateContext() argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 150 config.device_options = GetDeviceOptionsFromEnv(); in ExecutorForDevice()
159 config.device_options = GetDeviceOptionsFromEnv(); in ExecutorForDeviceWithPluginConfig()
173 auto init_status = executor->Init(config.ordinal, config.device_options); in GetUncachedExecutor()
|
| D | cuda_driver.cc | 369 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags() argument
374 if (device_options.flags() & DeviceOptions::kDoNotReclaimStackAllocation) { in DeviceOptionsToContextFlags()
380 if (device_options.flags() & DeviceOptions::kScheduleSpin) { in DeviceOptionsToContextFlags()
383 if (device_options.flags() & DeviceOptions::kScheduleYield) { in DeviceOptionsToContextFlags()
386 if (device_options.flags() & DeviceOptions::kScheduleBlockingSync) { in DeviceOptionsToContextFlags()
394 int device_ordinal, CUdevice device, const DeviceOptions& device_options, in CreateContext() argument
399 if (!DeviceOptionsToContextFlags(device_options, &flags)) { in CreateContext()
|
| D | cuda_gpu_executor.cc | 131 DeviceOptions device_options) { in Init() argument
144 status = GpuDriver::CreateContext(device_ordinal_, device_, device_options, in Init()
|
| D | BUILD | 99 "//tensorflow/stream_executor:device_options",
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 59 "//tensorflow/stream_executor:device_options",
|
| D | gpu_executor.h | 62 port::Status Init(int device_ordinal, DeviceOptions device_options) override;
|
| D | gpu_driver.h | 164 const DeviceOptions& device_options,
|