Home
last modified time | relevance | path

Searched refs:possible_devices (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcolocation_graph.cc835 PossibleDevices possible_devices; in AddHostOnlyDataTypesConstraints() local
837 std::back_inserter(possible_devices.device_types), in AddHostOnlyDataTypesConstraints()
841 possible_devices, /*allow_soft_placement=*/false)); in AddHostOnlyDataTypesConstraints()
996 const PossibleDevices& possible_devices = groups.group_devices[group_id]; in ApplyIOColocationGroups() local
997 TF_RETURN_IF_ERROR(LimitToPossibleDevices(*group_node, possible_devices)); in ApplyIOColocationGroups()
1108 std::vector<Device*>* possible_devices) { in GetSoftDeviceCandidates() argument
1115 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates()
1116 if (!possible_devices->empty()) { in GetSoftDeviceCandidates()
1117 *possible_devices = FilterSupportedDevices( in GetSoftDeviceCandidates()
1118 *possible_devices, root_member.supported_device_types(), in GetSoftDeviceCandidates()
[all …]
Dcolocation_graph.h102 const std::vector<Device*>& possible_devices() { return possible_devices_; } in possible_devices() function
254 const std::vector<Device*>** possible_devices);
348 std::vector<Device*>* possible_devices);
Dinspecting_placer.cc99 PossibleDevices& possible_devices = (*group_devices)[assigned_group_id]; in FillGroups() local
101 TF_RETURN_IF_ERROR(member.FillPossibleDevices(&possible_devices)); in FillGroups()