Searched refs:LogicalIdsForDevice (Results 1 – 4 of 4) sorted by relevance
45 StatusOr<std::pair<int, int>> DeviceAssignment::LogicalIdsForDevice( in LogicalIdsForDevice() function in xla::DeviceAssignment70 TF_ASSIGN_OR_RETURN(auto logical_ids, LogicalIdsForDevice(device_id)); in ReplicaIdForDevice()
54 StatusOr<std::pair<int, int>> LogicalIdsForDevice(
94 device_assignment.LogicalIdsForDevice(device_id)); in GetParticipatingDevices()
29 TF_ASSIGN_OR_RETURN(auto logical_ids, params.device_assn->LogicalIdsForDevice( in ExecuteOnStream()