| /external/tensorflow/tensorflow/stream_executor/ |
| D | executor_cache.cc | 48 iter.first.device_options == config.device_options) { in GetOrCreate()
90 iter.first.device_options == config.device_options) { in Get()
|
| D | BUILD | 63 "device_options.h",
117 "device_options.h",
308 name = "device_options",
309 hdrs = ["device_options.h"],
320 "device_options.h",
465 ":device_options",
554 "device_options.h",
603 "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 | 89 DeviceOptions device_options; member
|
| 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() argument
187 return implementation_->Init(device_ordinal_, std::move(device_options)); in Init()
|
| D | stream_executor_pimpl.h | 83 port::Status Init(DeviceOptions device_options);
|
| /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()
77 auto init_status = executor->Init(config.device_options); in GetUncachedExecutor()
|
| D | host_gpu_executor.cc | 49 DeviceOptions device_options) { in Init() argument
51 device_options.non_portable_tags.find("host_thread_stack_size_in_bytes"); in Init()
52 if (it != device_options.non_portable_tags.end()) { in Init()
|
| D | host_gpu_executor.h | 51 port::Status Init(int device_ordinal, DeviceOptions device_options) override;
|
| /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()
81 auto init_status = executor->Init(config.device_options); in GetUncachedExecutor()
|
| D | executor.h | 56 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()
140 auto init_status = executor->Init(config.device_options); in GetUncachedExecutor()
|
| D | BUILD | 56 "//tensorflow/stream_executor:device_options",
|
| D | rocm_gpu_executor.cc | 161 DeviceOptions device_options) { in Init() argument
174 status = GpuDriver::CreateContext(device_ordinal_, device_, device_options, in Init()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 153 config.device_options = GetDeviceOptionsFromEnv(); in ExecutorForDevice()
162 config.device_options = GetDeviceOptionsFromEnv(); in ExecutorForDeviceWithPluginConfig()
177 auto init_status = executor->Init(config.device_options); in GetUncachedExecutor()
|
| D | cuda_driver.cc | 366 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags() argument
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() argument
396 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()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | cpu_device.cc | 49 config.device_options.non_portable_tags["host_thread_stack_size_in_bytes"] = in GetCpuClient()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.cc | 40 ::stream_executor::DeviceOptions device_options) { in Init() argument
44 device_options.flags()); in Init()
|
| D | tpu_executor.h | 67 ::stream_executor::DeviceOptions device_options) override;
|
| D | tpu_executor_c_api.h | 46 SE_DeviceOptions* device_options, TF_Status* status);
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 70 "//tensorflow/stream_executor:device_options",
|
| D | gpu_executor.h | 62 port::Status Init(int device_ordinal, DeviceOptions device_options) override;
|
| D | gpu_driver.h | 221 const DeviceOptions& device_options,
|