| /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
71 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 | 89 DeviceOptions device_options;
|
| D | stream_executor_internal.h | 179 DeviceOptions device_options) = 0;
|
| D | stream_executor_pimpl.cc | 186 port::Status StreamExecutor::Init(DeviceOptions device_options) { in Init()
190 port::Status StreamExecutor::Init() { return Init(DeviceOptions::Default()); } in Init()
|
| D | stream_executor_pimpl.h | 83 port::Status Init(DeviceOptions device_options);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 41 const DeviceOptions GetDeviceOptionsFromEnv() { in GetDeviceOptionsFromEnv()
46 return DeviceOptions::Default(); in GetDeviceOptionsFromEnv()
51 device_flags = DeviceOptions::kScheduleSpin; in GetDeviceOptionsFromEnv()
53 device_flags = DeviceOptions::kScheduleYield; in GetDeviceOptionsFromEnv()
55 device_flags = DeviceOptions::kScheduleBlockingSync; in GetDeviceOptionsFromEnv()
64 return DeviceOptions(device_flags); in GetDeviceOptionsFromEnv()
|
| D | cuda_driver.cc | 366 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags()
368 static_assert(DeviceOptions::kMask == 0xf, in DeviceOptionsToContextFlags()
371 if (device_options.flags() & DeviceOptions::kDoNotReclaimStackAllocation) { in DeviceOptionsToContextFlags()
377 if (device_options.flags() & DeviceOptions::kScheduleSpin) { in DeviceOptionsToContextFlags()
380 if (device_options.flags() & DeviceOptions::kScheduleYield) { in DeviceOptionsToContextFlags()
383 if (device_options.flags() & DeviceOptions::kScheduleBlockingSync) { in DeviceOptionsToContextFlags()
391 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 | 53 config.device_options = DeviceOptions::Default(); in ExecutorForDevice()
62 config.device_options = DeviceOptions::Default(); in ExecutorForDeviceWithPluginConfig()
|
| D | host_gpu_executor.h | 51 port::Status Init(int device_ordinal, DeviceOptions device_options) override;
|
| D | host_gpu_executor.cc | 49 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.cc | 55 config.device_options = DeviceOptions::Default(); in ExecutorForDevice()
65 config.device_options = DeviceOptions::Default(); in ExecutorForDeviceWithPluginConfig()
|
| D | executor.h | 56 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 | 345 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags()
347 static_assert(DeviceOptions::kMask == 0xf, in DeviceOptionsToContextFlags()
353 int device_ordinal, hipDevice_t device, const DeviceOptions& device_options, in CreateContext()
|
| D | rocm_gpu_executor.cc | 161 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_defn.h | 63 stream_executor::DeviceOptions options;
|
| D | tpu_executor.h | 67 ::stream_executor::DeviceOptions device_options) override;
|
| D | tpu_executor.cc | 40 ::stream_executor::DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 62 port::Status Init(int device_ordinal, DeviceOptions device_options) override;
|
| D | gpu_driver.h | 221 const DeviceOptions& device_options,
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 257 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|