Searched defs:device_options (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | platform.h | 92 DeviceOptions device_options; member
|
| D | stream_executor_pimpl.cc | 186 port::Status StreamExecutor::Init(DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_factory.cc | 79 const GPUOptions& device_options, in SingleVirtualDeviceMemoryLimit()
176 const auto& device_options = options.config.gpu_options(); in CreateDevices() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 55 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | host_gpu_executor.cc | 49 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_executor.cc | 40 ::stream_executor::DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 345 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags()
353 int device_ordinal, hipDevice_t device, const DeviceOptions& device_options, in CreateContext()
|
| D | rocm_gpu_executor.cc | 163 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_driver.cc | 303 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, in DeviceOptionsToContextFlags()
328 int device_ordinal, CUdevice device, const DeviceOptions& device_options, in CreateContext()
|
| D | cuda_gpu_executor.cc | 129 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 229 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|