Searched defs:device_options (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | platform.h | 89 DeviceOptions device_options; member
|
| D | stream_executor_pimpl.cc | 186 port::Status StreamExecutor::Init(DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.h | 56 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 49 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/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 | 161 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 366 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, 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/c/experimental/stream_executor/ |
| D | stream_executor.cc | 257 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|