Searched refs:LocalDevice (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | local_device.cc | 53 bool LocalDevice::use_global_threadpool_ = true;
54 mutex LocalDevice::global_tp_mu_;
55 gtl::InlinedVector<LocalDevice::EigenThreadPoolInfo*, 4>
56 LocalDevice::global_tp_info_;
58 struct LocalDevice::EigenThreadPoolInfo {
114 LocalDevice::LocalDevice(const SessionOptions& options, in LocalDevice() function in tensorflow::LocalDevice
120 LocalDevice::EigenThreadPoolInfo* tp_info; in LocalDevice()
138 global_tp_info_[numa_node] = new LocalDevice::EigenThreadPoolInfo( in LocalDevice()
144 global_tp_info_.push_back(new LocalDevice::EigenThreadPoolInfo( in LocalDevice()
153 owned_tp_info_.reset(new LocalDevice::EigenThreadPoolInfo( in LocalDevice()
[all …]
|
| D | local_device.h | 34 class LocalDevice : public Device {
36 LocalDevice(const SessionOptions& options,
38 ~LocalDevice() override;
59 TF_DISALLOW_COPY_AND_ASSIGN(LocalDevice);
|
| D | threadpool_device.h | 26 class ThreadPoolDevice : public LocalDevice {
|
| D | kernel_benchmark_testlib.cc | 67 LocalDevice::set_use_global_threadpool(false); in Benchmark()
|
| D | threadpool_device.cc | 60 : LocalDevice(options, Device::BuildDeviceAttributes( in ThreadPoolDevice()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.h | 47 class XlaCompilationDevice : public LocalDevice {
|
| D | xla_compilation_device.cc | 73 : LocalDevice(options, Device::BuildDeviceAttributes( in XlaCompilationDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device.h | 44 class PluggableDevice : public LocalDevice {
|
| D | pluggable_device.cc | 145 : LocalDevice(options, Device::BuildDeviceAttributes( in PluggableDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.h | 51 class XlaDevice : public LocalDevice {
|
| D | xla_device.cc | 202 : LocalDevice(session_options, in XlaDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 75 class BaseGPUDevice : public LocalDevice {
|
| D | gpu_device.cc | 393 : LocalDevice(options, Device::BuildDeviceAttributes(name, DEVICE_GPU, in BaseGPUDevice()
|