Searched refs:getFederatedTrainingTask (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/scheduling/ |
| D | FederatedComputeJobManagerTest.java | 190 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalledSuccess()
219 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalledFailureDueToFaultyOwnerPackage()
251 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalledFailureDueToTaskLimit()
278 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalled_firstTime()
345 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalledFLWithInterval()
374 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalled_multipleTimes_sameParams()
422 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalled_multipleTimes_sameParamsFLWithIntervalLargerThanServerMax()
465 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalled_multipleTimes_sameParamsFLWithIntervalDifferentMax()
500 taskList = mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalled_multipleTimes_sameParamsFLWithIntervalDifferentMax()
541 mTrainingTaskDao.getFederatedTrainingTask(null, null); in testOnTrainerStartCalled_fLCustomerSpecifiedIntervalSmallerThanDefinedMin()
[all …]
|
| /packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/data/ |
| D | FederatedTrainingTaskDaoTest.java | 107 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(2); in findAndRemoveTaskByJobId_success()
112 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(1); in findAndRemoveTaskByJobId_success()
132 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(2); in findAndRemoveTaskByPopulationNameAndJobId_success()
138 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(1); in findAndRemoveTaskByPopulationNameAndJobId_success()
176 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(2); in findAndRemoveTaskByPopulationName_success()
182 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(1); in findAndRemoveTaskByPopulationName_success()
196 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(2); in findAndRemoveTaskByPopulationNameAndCallingPackage_success()
203 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(1); in findAndRemoveTaskByPopulationNameAndCallingPackage_success()
245 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(4); in findAndRemoveTaskByPopulationNameAndAndOwnerId_success()
252 assertThat(mTrainingTaskDao.getFederatedTrainingTask(null, null)).hasSize(3); in findAndRemoveTaskByPopulationNameAndAndOwnerId_success()
|
| /packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/data/ |
| D | FederatedTrainingTaskDao.java | 115 public List<FederatedTrainingTask> getFederatedTrainingTask( in getFederatedTrainingTask() method in FederatedTrainingTaskDao
130 Iterables.getOnlyElement(getFederatedTrainingTask(selection, selectionArgs), null); in findAndRemoveTaskByJobId()
152 Iterables.getOnlyElement(getFederatedTrainingTask(selection, selectionArgs), null); in findAndRemoveTaskByPopulationName()
183 Iterables.getOnlyElement(getFederatedTrainingTask(selection, selectionArgs), null); in findAndRemoveTaskByPopulationNameAndCallingPackage()
220 Iterables.getOnlyElement(getFederatedTrainingTask(selection, selectionArgs), null); in findAndRemoveTaskByPopulationNameAndOwnerId()
253 Iterables.getOnlyElement(getFederatedTrainingTask(selection, selectionArgs), null); in findAndRemoveTaskByPopulationAndJobId()
|
| /packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/scheduling/ |
| D | FederatedComputeLearningJobScheduleOrchestrator.java | 90 mFederatedTrainingTaskDao.getFederatedTrainingTask(null, null); in checkAndSchedule()
|