Searched refs:default_local_device (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.h | 73 const Device* default_local_device, bool allow_soft_placement,
77 const Device* default_local_device);
|
| D | placer.cc | 136 const DeviceSet* devices, const Device* default_local_device, in Placer() argument
142 default_local_device_(default_local_device), in Placer()
147 const DeviceSet* devices, const Device* default_local_device) in Placer() argument
149 default_local_device, true, false) {} in Placer()
|
| D | colocation_graph.cc | 590 const Device* client_device, const Device* default_local_device) { in LocalAddressSpec() argument
595 if (default_local_device != nullptr) { in LocalAddressSpec()
596 return DeviceNameUtils::AddressSpace(default_local_device->parsed_name()); in LocalAddressSpec()
607 const Device* default_local_device, in ColocationGraph() argument
613 inspecting_placer_(stack, flib_def, device_set, default_local_device, in ColocationGraph()
619 LocalAddressSpec(device_set->client_device(), default_local_device)), in ColocationGraph()
620 default_local_device_(default_local_device), in ColocationGraph()
1481 const Device* default_local_device) { in FilterSupportedDevices() argument
1488 if (default_local_device && in FilterSupportedDevices()
1489 (device == default_local_device || in FilterSupportedDevices()
[all …]
|
| D | colocation_graph.h | 220 const Device* default_local_device, bool allow_soft_placement,
267 const Device* default_local_device);
|
| D | placer_test.cc | 260 Status Place(Graph* graph, DeviceSet* devices, Device* default_local_device, in Place() argument
262 Placer placer(graph, "", &graph->flib_def(), devices, default_local_device, in Place()
|