| /packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/data/ |
| D | FederatedTrainingTaskDaoTest.java | 101 FederatedTrainingTask task = createDefaultFederatedTrainingTask(); in findAndRemoveTaskByJobId_success()
104 FederatedTrainingTask task2 = in findAndRemoveTaskByJobId_success()
109 FederatedTrainingTask removedTask = mTrainingTaskDao.findAndRemoveTaskByJobId(JOB_ID); in findAndRemoveTaskByJobId_success()
117 FederatedTrainingTask removedTask = mTrainingTaskDao.findAndRemoveTaskByJobId(JOB_ID); in findAndRemoveTaskByJobId_nonExist()
124 FederatedTrainingTask task = createDefaultFederatedTrainingTask(); in findAndRemoveTaskByPopulationNameAndJobId_success()
126 FederatedTrainingTask task2 = in findAndRemoveTaskByPopulationNameAndJobId_success()
134 FederatedTrainingTask removedTask = in findAndRemoveTaskByPopulationNameAndJobId_success()
143 FederatedTrainingTask task = createDefaultFederatedTrainingTask(); in testGetNumberOfFederatedTrainingTasksPerOwnerPackage()
145 FederatedTrainingTask task2 = in testGetNumberOfFederatedTrainingTasksPerOwnerPackage()
160 FederatedTrainingTask removedTask = in findAndRemoveTaskByPopulationNameAndJobId_nonExist()
[all …]
|
| D | FederatedTrainingTaskTest.java | 84 FederatedTrainingTask task = createFederatedTrainingTaskWithAllFields(); in readAndWrite()
90 FederatedTrainingTask storedFederatedTrainingTask = in readAndWrite()
92 FederatedTrainingTask.readFederatedTrainingTasksFromDatabase( in readAndWrite()
100 FederatedTrainingTask task = createFederatedTrainingTaskWithRequiredFields(); in readAndWrite_onlyRequiredFields()
106 FederatedTrainingTask storedFederatedTrainingTask = in readAndWrite_onlyRequiredFields()
108 FederatedTrainingTask.readFederatedTrainingTasksFromDatabase( in readAndWrite_onlyRequiredFields()
126 FederatedTrainingTask task1 = createFederatedTrainingTaskWithAllFields(); in queryWithJobId()
128 FederatedTrainingTask task2 = in queryWithJobId()
132 List<FederatedTrainingTask> trainingTaskList = in queryWithJobId()
133 FederatedTrainingTask.readFederatedTrainingTasksFromDatabase( in queryWithJobId()
[all …]
|
| D | DataTestUtil.java | 25 public static boolean isEqualTask(FederatedTrainingTask task1, FederatedTrainingTask task2) { in isEqualTask()
|
| /packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/scheduling/ |
| D | FederatedComputeJobManager.java | 37 import com.android.federatedcompute.services.data.FederatedTrainingTask;
148 TrainingOptions newTaskOptions, FederatedTrainingTask existingTask) { in trainingIntervalChanged()
169 FederatedTrainingTask existingTask = in onTrainerStartCalled()
175 Set<FederatedTrainingTask> trainingTasksToCancel = new HashSet<>(); in onTrainerStartCalled()
186 FederatedTrainingTask newTask; in onTrainerStartCalled()
204 FederatedTrainingTask.Builder newTaskBuilder = in onTrainerStartCalled()
205 FederatedTrainingTask.builder() in onTrainerStartCalled()
235 FederatedTrainingTask.Builder newTaskBuilder = in onTrainerStartCalled()
315 for (FederatedTrainingTask trainingTaskToCancel : trainingTasksToCancel) { in onTrainerStartCalled()
335 FederatedTrainingTask taskToCancel =
[all …]
|
| D | JobSchedulerHelper.java | 25 import com.android.federatedcompute.services.data.FederatedTrainingTask;
43 public boolean scheduleTask(Context context, FederatedTrainingTask newTask) { in scheduleTask()
54 public void cancelTask(Context context, FederatedTrainingTask taskToCancel) { in cancelTask()
86 private JobInfo convertToJobInfo(Context context, FederatedTrainingTask task) { in convertToJobInfo()
107 public boolean isTaskScheduled(Context context, FederatedTrainingTask task) { in isTaskScheduled()
|
| D | FederatedComputeLearningJobScheduleOrchestrator.java | 24 import com.android.federatedcompute.services.data.FederatedTrainingTask;
89 List<FederatedTrainingTask> tasks = in checkAndSchedule()
92 for (FederatedTrainingTask task : tasks) { in checkAndSchedule()
|
| D | SchedulingUtil.java | 27 import com.android.federatedcompute.services.data.FederatedTrainingTask;
118 FederatedTrainingTask existingTask, in getEarliestRuntimeForExistingTask()
|
| /packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/data/ |
| D | FederatedTrainingTaskDao.java | 96 public boolean updateOrInsertFederatedTrainingTask(FederatedTrainingTask trainingTask) { in updateOrInsertFederatedTrainingTask()
115 public List<FederatedTrainingTask> getFederatedTrainingTask( in getFederatedTrainingTask()
121 return FederatedTrainingTask.readFederatedTrainingTasksFromDatabase( in getFederatedTrainingTask()
126 public FederatedTrainingTask findAndRemoveTaskByJobId(int jobId) { in findAndRemoveTaskByJobId()
129 FederatedTrainingTask task = in findAndRemoveTaskByJobId()
148 public FederatedTrainingTask findAndRemoveTaskByPopulationName(String populationName) { in findAndRemoveTaskByPopulationName()
151 FederatedTrainingTask task = in findAndRemoveTaskByPopulationName()
174 public FederatedTrainingTask findAndRemoveTaskByPopulationNameAndCallingPackage( in findAndRemoveTaskByPopulationNameAndCallingPackage()
182 FederatedTrainingTask task = in findAndRemoveTaskByPopulationNameAndCallingPackage()
207 public FederatedTrainingTask findAndRemoveTaskByPopulationNameAndOwnerId( in findAndRemoveTaskByPopulationNameAndOwnerId()
[all …]
|
| D | FederatedTrainingTask.java | 39 public abstract class FederatedTrainingTask { class
40 private static final String TAG = FederatedTrainingTask.class.getSimpleName();
224 public abstract FederatedTrainingTask build(); in build()
281 static List<FederatedTrainingTask> readFederatedTrainingTasksFromDatabase( in readFederatedTrainingTasksFromDatabase()
283 List<FederatedTrainingTask> taskList = new ArrayList<>(); in readFederatedTrainingTasksFromDatabase()
319 FederatedTrainingTask.Builder trainingTaskBuilder = in readFederatedTrainingTasksFromDatabase()
320 FederatedTrainingTask.builder() in readFederatedTrainingTasksFromDatabase()
|
| /packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/scheduling/ |
| D | FederatedComputeJobManagerTest.java | 46 import com.android.federatedcompute.services.data.FederatedTrainingTask;
189 List<FederatedTrainingTask> taskList = in testOnTrainerStartCalledSuccess()
218 List<FederatedTrainingTask> taskList = in testOnTrainerStartCalledFailureDueToFaultyOwnerPackage()
250 List<FederatedTrainingTask> taskList = in testOnTrainerStartCalledFailureDueToTaskLimit()
277 List<FederatedTrainingTask> taskList = in testOnTrainerStartCalled_firstTime()
337 FederatedTrainingTask expectedTask = in testOnTrainerStartCalledFLWithInterval()
344 List<FederatedTrainingTask> taskList = in testOnTrainerStartCalledFLWithInterval()
373 List<FederatedTrainingTask> taskList = in testOnTrainerStartCalled_multipleTimes_sameParams()
375 FederatedTrainingTask expectedTask = in testOnTrainerStartCalled_multipleTimes_sameParams()
421 List<FederatedTrainingTask> taskList = in testOnTrainerStartCalled_multipleTimes_sameParamsFLWithIntervalLargerThanServerMax()
[all …]
|
| D | JobSchedulerHelperTest.java | 33 import com.android.federatedcompute.services.data.FederatedTrainingTask;
69 private static final FederatedTrainingTask TRAINING_TASK =
70 FederatedTrainingTask.builder()
119 FederatedTrainingTask trainingTask = in scheduleTask_overrideIdleConstraint_waivesIdle()
139 FederatedTrainingTask trainingTask = in scheduleTask_overrideUnmeteredNetworkConstraint_waivesNetwork()
159 FederatedTrainingTask trainingTask = in scheduleTask_overrideChargingConstraint_waivesCharging()
|
| D | FederatedComputeLearningJobScheduleOrchestratorTest.java | 36 import com.android.federatedcompute.services.data.FederatedTrainingTask;
89 FederatedTrainingTask task1 = in checkAndScheduleTest()
90 FederatedTrainingTask.builder() in checkAndScheduleTest()
105 FederatedTrainingTask task2 = in checkAndScheduleTest()
106 FederatedTrainingTask.builder() in checkAndScheduleTest()
|
| /packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/training/ |
| D | EligibilityDecider.java | 32 import com.android.federatedcompute.services.data.FederatedTrainingTask;
68 FederatedTrainingTask task, in computeEligibility()
150 FederatedTrainingTask task, in computePerTaskDataAvailability()
|
| D | ResultCallbackHelper.java | 31 import com.android.federatedcompute.services.data.FederatedTrainingTask;
70 String taskId, FederatedTrainingTask task, ComputationResult result) { in callHandleResult()
|
| D | FederatedComputeWorker.java | 59 import com.android.federatedcompute.services.data.FederatedTrainingTask;
204 FederatedTrainingTask trainingTask,
234 private FederatedTrainingTask getTrainableTask(
238 FederatedTrainingTask trainingTask = mJobManager.onTrainingStarted(jobId);
1180 private final FederatedTrainingTask mTask;
1196 FederatedTrainingTask task,
|
| /packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/training/ |
| D | ResultCallbackHelperTest.java | 38 import com.android.federatedcompute.services.data.FederatedTrainingTask;
73 private static final FederatedTrainingTask TRAINING_TASK =
74 FederatedTrainingTask.builder()
|
| D | EligibilityDeciderTest.java | 48 import com.android.federatedcompute.services.data.FederatedTrainingTask;
445 private FederatedTrainingTask createDefaultFederatedTrainingTask() { in createDefaultFederatedTrainingTask()
446 return FederatedTrainingTask.builder() in createDefaultFederatedTrainingTask()
|
| D | FederatedComputeWorkerTest.java | 69 import com.android.federatedcompute.services.data.FederatedTrainingTask;
271 private static final FederatedTrainingTask FEDERATED_TRAINING_TASK_1 =
272 FederatedTrainingTask.builder()
|
| /packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/examplestore/ |
| D | ExampleStoreServiceProvider.java | 36 import com.android.federatedcompute.services.data.FederatedTrainingTask;
69 FederatedTrainingTask task, in getExampleIterator()
|