Searched refs:DeviceOptions (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_options.h | 36 struct DeviceOptions { struct
57 explicit DeviceOptions(unsigned flags) : flags_(flags) { in DeviceOptions() function
62 static DeviceOptions Default() { return DeviceOptions(0); } in Default() argument
66 bool operator==(const DeviceOptions& other) const { argument
70 bool operator!=(const DeviceOptions& other) const {
|
| 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;
|
| D | stream_executor_internal.h | 180 DeviceOptions device_options) = 0;
|
| D | stream_executor_pimpl.cc | 225 DeviceOptions device_options) { in Init()
231 return Init(0, DeviceOptions::Default()); in Init()
|
| D | stream_executor_pimpl.h | 83 port::Status Init(int device_ordinal, DeviceOptions device_options);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 38 const DeviceOptions GetDeviceOptionsFromEnv() { in GetDeviceOptionsFromEnv()
43 return DeviceOptions::Default(); in GetDeviceOptionsFromEnv()
48 device_flags = DeviceOptions::kScheduleSpin; in GetDeviceOptionsFromEnv()
50 device_flags = DeviceOptions::kScheduleYield; in GetDeviceOptionsFromEnv()
52 device_flags = DeviceOptions::kScheduleBlockingSync; in GetDeviceOptionsFromEnv()
61 return DeviceOptions(device_flags); in GetDeviceOptionsFromEnv()
|
| D | cuda_driver.cc | 369 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags()
371 static_assert(DeviceOptions::kMask == 0xf, in DeviceOptionsToContextFlags()
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()
|
| D | cuda_gpu_executor.cc | 131 DeviceOptions device_options) { in Init()
|
| /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()
|
| D | host_gpu_executor.h | 49 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|
| /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()
|
| D | executor.h | 57 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|
| /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()
|
| D | rocm_driver.cc | 355 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags()
357 static_assert(DeviceOptions::kMask == 0xf, in DeviceOptionsToContextFlags()
363 int device_ordinal, hipDevice_t device, const DeviceOptions& device_options, in CreateContext()
|
| D | rocm_gpu_executor.cc | 144 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 164 const DeviceOptions& device_options,
|
| D | gpu_executor.h | 62 port::Status Init(int device_ordinal, DeviceOptions device_options) override;
|