Searched refs:args_device_selector (Results 1 – 1 of 1) sorted by relevance
1968 TensorDevicePlacer args_device_selector(num_cores_per_replica, arg_types, in AssignArgsAndRetvalsToCores() local2058 assigned_core = args_device_selector.RetrieveAssignment(i); in AssignArgsAndRetvalsToCores()2078 args_device_selector.ReportDeviceAssigned(*assigned_core, i); in AssignArgsAndRetvalsToCores()2085 args_device_selector.ReportDeviceAssigned(core, i); in AssignArgsAndRetvalsToCores()2096 args_device_selector.ReportDeviceAssigned(core, i); in AssignArgsAndRetvalsToCores()