Home
last modified time | relevance | path

Searched refs:has_device_cpu (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer.cc269 bool has_device_cpu, const string& device) { in TryFindHostDevice() argument
271 if (device.empty() && has_device_cpu) { in TryFindHostDevice()
324 const bool has_device_cpu = devices.find("/device:CPU:0") != devices.end(); in Optimize() local
343 internal::TryFindHostDevice(devices, has_device_cpu, node.device()); in Optimize()
Dpin_to_host_optimizer.h30 bool has_device_cpu, const string& device);