Searched refs:LocalDevice (Results 1 – 12 of 12) 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 {
121 LocalDevice::LocalDevice(const SessionOptions& options, in LocalDevice() function in tensorflow::LocalDevice
127 LocalDevice::EigenThreadPoolInfo* tp_info; in LocalDevice()
145 global_tp_info_[numa_node] = new LocalDevice::EigenThreadPoolInfo( in LocalDevice()
151 global_tp_info_.push_back(new LocalDevice::EigenThreadPoolInfo( in LocalDevice()
160 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 | 25 class ThreadPoolDevice : public LocalDevice {
|
| D | threadpool_device.cc | 48 : LocalDevice(options, Device::BuildDeviceAttributes( in ThreadPoolDevice()
|
| D | kernel_benchmark_testlib.cc | 57 LocalDevice::set_use_global_threadpool(false); in Benchmark()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.h | 196 class SYCLDevice : public LocalDevice {
202 : LocalDevice(options, Device::BuildDeviceAttributes( in SYCLDevice()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.h | 47 class XlaCompilationDevice : public LocalDevice {
|
| D | xla_compilation_device.cc | 77 : LocalDevice(options, Device::BuildDeviceAttributes( in XlaCompilationDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.h | 50 class XlaDevice : public LocalDevice {
|
| D | xla_device.cc | 191 : LocalDevice(session_options, in XlaDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 52 class BaseGPUDevice : public LocalDevice {
|
| D | gpu_device.cc | 304 : LocalDevice(options, Device::BuildDeviceAttributes(name, DEVICE_GPU, in BaseGPUDevice()
|