Searched refs:cpu_device_ (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 47 cpu_device_ = devices.back().get();
60 Device* cpu_device() { return cpu_device_; } in cpu_device()
67 Device* cpu_device_; member in tensorflow::__anone5cc3cd90111::TestEnv
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 127 MK_OPT("constfold", new ConstantFolding(cpu_device_)); in MakeNewOptimizer()
134 MK_OPT("loop", new LoopOptimizer(cfg_.loop_optimization(), cpu_device_)); in MakeNewOptimizer()
149 : cpu_device_(cpu_device), in MetaOptimizer()
152 DCHECK(cpu_device_ == nullptr || in MetaOptimizer()
153 cpu_device_->attributes().device_type() == "CPU"); in MetaOptimizer()
176 MakeUnique<ConstantFolding>(cfg_.constant_folding(), cpu_device_)); in InitializeOptimizers()
193 MakeUnique<LoopOptimizer>(cfg_.loop_optimization(), cpu_device_)); in InitializeOptimizers()
|
| D | loop_optimizer.h | 68 DeviceBase* cpu_device_; variable
|
| D | meta_optimizer.h | 79 DeviceBase* const cpu_device_; // may be NULL variable
|
| D | loop_optimizer.cc | 699 cpu_device_(nullptr), in LoopOptimizer()
705 cpu_device_(cpu_device), in LoopOptimizer()
758 cpu_device_, resource_mgr_.get(), in RemoveDeadBranches()
|
| D | constant_folding.h | 244 DeviceBase* cpu_device_; variable
|
| D | constant_folding.cc | 229 : opt_level_(opt_level), cpu_device_(cpu_device) { in ConstantFolding()
1158 return ::tensorflow::grappler::EvaluateNode(node, inputs, cpu_device_, in EvaluateNode()
3433 if (cpu_device_ == nullptr) { in Optimize()
3435 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
|