Searched defs:device_options (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | platform.h | 87 DeviceOptions device_options; member
|
| D | stream_executor_pimpl.cc | 225 DeviceOptions device_options) { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| 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 | executor.h | 57 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 355 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, 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/cuda/ |
| D | cuda_driver.cc | 369 bool DeviceOptionsToContextFlags(const DeviceOptions& device_options, 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()
|