Searched refs:LocalDevice (Results 1 – 11 of 11) sorted by relevance
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::LocalDevice120 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 …]
34 class LocalDevice : public Device {36 LocalDevice(const SessionOptions& options,38 ~LocalDevice() override;59 TF_DISALLOW_COPY_AND_ASSIGN(LocalDevice);
25 class ThreadPoolDevice : public LocalDevice {
49 : LocalDevice(options, Device::BuildDeviceAttributes( in ThreadPoolDevice()
68 LocalDevice::set_use_global_threadpool(false); in Benchmark()
47 class XlaCompilationDevice : public LocalDevice {
73 : LocalDevice(options, Device::BuildDeviceAttributes( in XlaCompilationDevice()
50 class XlaDevice : public LocalDevice {
197 : LocalDevice(session_options, in XlaDevice()
52 class BaseGPUDevice : public LocalDevice {
385 : LocalDevice(options, Device::BuildDeviceAttributes(name, DEVICE_GPU, in BaseGPUDevice()