Searched refs:root_member (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 1102 Member& root_member = members_[root]; in LimitToAssignedDevice() local
1103 return root_member.AssignDevice(node); in LimitToAssignedDevice()
1107 const Node& node, const Member& root_member, int root_id, in GetSoftDeviceCandidates() argument
1114 root_member.GetPreferredSoftDeviceName(); in GetSoftDeviceCandidates()
1118 *possible_devices, root_member.supported_device_types(), in GetSoftDeviceCandidates()
1137 soft_device_name = root_member.GetSoftDeviceName(); in GetSoftDeviceCandidates()
1141 *possible_devices, root_member.supported_device_types(), in GetSoftDeviceCandidates()
1160 Member& root_member = members_[root]; in LimitToPossibleDevices() local
1161 return root_member.LimitToPossibleDevices(devices, allow_soft_placement_); in LimitToPossibleDevices()
1173 Member& root_member = members_[node_root]; in GetDevicesForNode() local
[all …]
|
| D | colocation_graph.h | 346 void GetSoftDeviceCandidates(const Node& node, const Member& root_member,
|