Home
last modified time | relevance | path

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

/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/scheduling/
DFederatedComputeJobManager.java218 .rescheduleCount(0) in onTrainerStartCalled()
284 newTaskBuilder.rescheduleCount(0); in onTrainerStartCalled()
480 int rescheduleCount = existingTask.rescheduleCount() + 1; in rescheduleFederatedTaskAfterTraining() local
481 if (checkRescheduleLimitsExceeded(intervalOptions, rescheduleCount)) { in rescheduleFederatedTaskAfterTraining()
490 newTaskBuilder.rescheduleCount(rescheduleCount); in rescheduleFederatedTaskAfterTraining()
493 newTaskBuilder.rescheduleCount(0); in rescheduleFederatedTaskAfterTraining()
501 TrainingIntervalOptions intervalOptions, int rescheduleCount) { in checkRescheduleLimitsExceeded() argument
504 return rescheduleCount > mFlags.getFcpRescheduleLimit(); in checkRescheduleLimitsExceeded()
507 return rescheduleCount > mFlags.getFcpRecurrentRescheduleLimit(); in checkRescheduleLimitsExceeded()
510 return rescheduleCount > mFlags.getFcpRescheduleLimit(); in checkRescheduleLimitsExceeded()
/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/data/
DFederatedTrainingTask.java164 public abstract int rescheduleCount(); in rescheduleCount() method in FederatedTrainingTask
220 public abstract Builder rescheduleCount(int rescheduleCount); in rescheduleCount() argument
237 return new AutoValue_FederatedTrainingTask.Builder().rescheduleCount(0); in builder()
271 values.put(FederatedTrainingTaskColumns.RESCHEDULE_COUNT, rescheduleCount()); in addToDatabase()
380 .rescheduleCount( in readFederatedTrainingTasksFromDatabase()
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/data/
DFederatedTrainingTaskTest.java121 assertThat(storedFederatedTrainingTask.rescheduleCount()).isEqualTo(0); in readAndWrite_onlyRequiredFields()
157 assertThat(trainingTask.rescheduleCount()).isEqualTo(RESCHEDULE_COUNT); in buildFederatedTrainingTask()
191 .rescheduleCount(RESCHEDULE_COUNT) in createFederatedTrainingTaskWithAllFields()
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/scheduling/
DFederatedComputeJobManagerTest.java765 assertThat(taskInDb.rescheduleCount()).isEqualTo(1); in testOnTrainerStartCalled_existingTask_rescheduleCountClear()
774 assertThat(taskInDb.rescheduleCount()).isEqualTo(0); in testOnTrainerStartCalled_existingTask_rescheduleCountClear()
929 .rescheduleCount(1) in testRescheduleFLTask_didnotContribute_oneOff()
1166 .rescheduleCount(1) in testRescheduleFLTask_recurrent_didnotContribute()
1321 .rescheduleCount(1) in testRescheduleFLTask_recurrent_rescheduleCountReset()
1365 .rescheduleCount(0) in testRescheduleFLTask_recurrent_rescheduleCountReset()
1538 .rescheduleCount(0); in basicFLTrainingTaskBuilder()