Searched refs:cpu_device_ (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 65 cpu_device_ = DeviceNameUtils::ParsedNameToString(parsed_name); in RecvAtHostOp()
71 VLOG(2) << " cpu_device_ = " << cpu_device_; in RecvAtHostOp()
91 ctx, DeviceNameUtils::ParseFullName(cpu_device_, &parsed_name), in ComputeAsync()
97 VLOG(2) << " cpu_device_ = " << cpu_device_; in ComputeAsync()
130 /*src_incarnation=*/1, cpu_device_, in ComputeAsync()
173 string cpu_device_; member in tensorflow::__anon2abb64b70111::RecvAtHostOp
222 cpu_device_ = DeviceNameUtils::ParsedNameToString(parsed_name); in SendFromHostOp()
228 VLOG(2) << " cpu_device_ = " << cpu_device_; in SendFromHostOp()
246 DeviceNameUtils::ParseFullName(cpu_device_, &parsed_name), in Compute()
252 VLOG(2) << " cpu_device_ = " << cpu_device_; in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | dynamic_device_mgr.cc | 29 DynamicDeviceMgr::DynamicDeviceMgr() : cpu_device_(nullptr) {} in DynamicDeviceMgr()
156 if (d == cpu_device_) { in RemoveDevices()
200 if (dynamic_devices_.find(cpu_device_) != dynamic_devices_.end()) { in HostCPU()
201 return cpu_device_; in HostCPU()
203 cpu_device_ = nullptr; in HostCPU()
207 cpu_device_ = pair.first; in HostCPU()
211 return cpu_device_; in HostCPU()
|
| D | device_mgr.cc | 34 cpu_device_(nullptr) { in StaticDeviceMgr()
49 if (cpu_device_ == nullptr && t == "CPU" && d->parsed_name().id == 0) { in StaticDeviceMgr()
50 cpu_device_ = d.get(); in StaticDeviceMgr()
155 Device* StaticDeviceMgr::HostCPU() const { return cpu_device_; } in HostCPU()
|
| D | device_mgr.h | 109 Device* cpu_device_; variable
161 mutable Device* cpu_device_ TF_GUARDED_BY(devices_mu_);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 48 cpu_device_ = devices.back().get();
62 Device* cpu_device() { return cpu_device_; } in cpu_device()
69 Device* cpu_device_; member in tensorflow::__anon9d2e75500111::TestEnv
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 191 cpu_device_, in MakeNewOptimizer()
208 MK_OPT("loop", new LoopOptimizer(cfg_.loop_optimization(), cpu_device_)); in MakeNewOptimizer()
223 : cpu_device_(cpu_device), in MetaOptimizer()
226 DCHECK(cpu_device_ == nullptr || in MetaOptimizer()
227 cpu_device_->attributes().device_type() == "CPU"); in MetaOptimizer()
256 cfg_.constant_folding(), cpu_device_, in InitializeOptimizers()
288 MakeUnique<LoopOptimizer>(cfg_.loop_optimization(), cpu_device_)); in InitializeOptimizers()
|
| D | loop_optimizer.h | 72 DeviceBase* cpu_device_; variable
|
| D | meta_optimizer.h | 92 DeviceBase* const cpu_device_; // may be NULL variable
|
| D | loop_optimizer.cc | 721 cpu_device_(nullptr), in LoopOptimizer()
727 cpu_device_(cpu_device), in LoopOptimizer()
786 cpu_device_, resource_mgr_.get(), in RemoveDeadBranches()
|
| D | constant_folding.h | 330 DeviceBase* cpu_device_; variable
|
| D | constant_folding.cc | 194 cpu_device_(cpu_device), in ConstantFolding()
1318 return ::tensorflow::grappler::EvaluateNode(node, inputs, cpu_device_, in EvaluateNode()
3986 if (cpu_device_ == nullptr) { in Optimize()
3988 cpu_device_ = owned_device_.get(); in Optimize()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_device.cu | 46 const Eigen::DefaultDevice& device() const { return cpu_device_; } in device()
64 Eigen::DefaultDevice cpu_device_; member
|