Home
last modified time | relevance | path

Searched refs:trainingTask (Results 1 – 4 of 4) sorted by relevance

/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/data/
DFederatedTrainingTaskTest.java143 FederatedTrainingTask trainingTask = createFederatedTrainingTaskWithAllFields(); in buildFederatedTrainingTask() local
145 assertThat(trainingTask.jobId()).isEqualTo(JOB_ID); in buildFederatedTrainingTask()
146 assertThat(trainingTask.appPackageName()).isEqualTo(PACKAGE_NAME); in buildFederatedTrainingTask()
147 assertThat(trainingTask.populationName()).isEqualTo(POPULATION_NAME); in buildFederatedTrainingTask()
148 assertThat(trainingTask.serverAddress()).isEqualTo(SERVER_ADDRESS); in buildFederatedTrainingTask()
149 assertThat(trainingTask.constraints()).isEqualTo(TRAINING_CONSTRAINTS); in buildFederatedTrainingTask()
150 assertThat(trainingTask.intervalOptions()).isEqualTo(INTERVAL_OPTIONS); in buildFederatedTrainingTask()
151 assertThat(trainingTask.creationTime()).isEqualTo(CREATION_TIME); in buildFederatedTrainingTask()
152 assertThat(trainingTask.lastScheduledTime()).isEqualTo(LAST_SCHEDULE_TIME); in buildFederatedTrainingTask()
153 assertThat(trainingTask.lastRunStartTime()).isEqualTo(LAST_RUN_START_TIME); in buildFederatedTrainingTask()
[all …]
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/scheduling/
DJobSchedulerHelperTest.java119 FederatedTrainingTask trainingTask = in scheduleTask_overrideIdleConstraint_waivesIdle() local
128 assertThat(mJobSchedulerHelper.scheduleTask(mContext, trainingTask)).isTrue(); in scheduleTask_overrideIdleConstraint_waivesIdle()
139 FederatedTrainingTask trainingTask = in scheduleTask_overrideUnmeteredNetworkConstraint_waivesNetwork() local
148 assertThat(mJobSchedulerHelper.scheduleTask(mContext, trainingTask)).isTrue(); in scheduleTask_overrideUnmeteredNetworkConstraint_waivesNetwork()
159 FederatedTrainingTask trainingTask = in scheduleTask_overrideChargingConstraint_waivesCharging() local
168 assertThat(mJobSchedulerHelper.scheduleTask(mContext, trainingTask)).isTrue(); in scheduleTask_overrideChargingConstraint_waivesCharging()
/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/training/
DFederatedComputeWorker.java204 FederatedTrainingTask trainingTask,
210 TrainingRun run = new TrainingRun(jobId, trainingTask, trainingEventLogger, callback);
238 FederatedTrainingTask trainingTask = mJobManager.onTrainingStarted(jobId);
239 if (trainingTask == null) {
244 trainingEventLogger.setPopulationName(trainingTask.populationName());
245 String taskCallingPackageName = trainingTask.ownerPackageName();
249 if (!checkTrainingConditions(trainingTask.getTrainingConstraints())) {
255 trainingTask.populationName(),
256 trainingTask.getTrainingIntervalOptions(),
269 trainingTask.populationName());
[all …]
/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/data/
DFederatedTrainingTaskDao.java96 public boolean updateOrInsertFederatedTrainingTask(FederatedTrainingTask trainingTask) { in updateOrInsertFederatedTrainingTask() argument
102 return trainingTask.addToDatabase(db); in updateOrInsertFederatedTrainingTask()
108 trainingTask.populationName()); in updateOrInsertFederatedTrainingTask()