Searched refs:retvals_device_selector (Results 1 – 1 of 1) sorted by relevance
2115 TensorDevicePlacer retvals_device_selector(num_cores_per_replica, in AssignArgsAndRetvalsToCores() local2171 assigned_core = retvals_device_selector.RetrieveAssignment(i); in AssignArgsAndRetvalsToCores()2184 retvals_device_selector.ReportDeviceAssigned(*assigned_core, i); in AssignArgsAndRetvalsToCores()2190 retvals_device_selector.ReportDeviceAssigned(core, i); in AssignArgsAndRetvalsToCores()2201 retvals_device_selector.ReportDeviceAssigned(core, i); in AssignArgsAndRetvalsToCores()