Home
last modified time | relevance | path

Searched refs:FederatedTrainingTaskDao (Results 1 – 14 of 14) sorted by relevance

/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/data/
DFederatedTrainingTaskDao.java43 public class FederatedTrainingTaskDao { class
45 private static final String TAG = FederatedTrainingTaskDao.class.getSimpleName();
48 private static volatile FederatedTrainingTaskDao sSingletonInstance;
50 private FederatedTrainingTaskDao(FederatedComputeDbHelper dbHelper) { in FederatedTrainingTaskDao() method in FederatedTrainingTaskDao
56 public static FederatedTrainingTaskDao getInstance(Context context) { in getInstance()
58 synchronized (FederatedTrainingTaskDao.class) { in getInstance()
61 new FederatedTrainingTaskDao( in getInstance()
71 public static FederatedTrainingTaskDao getInstanceForTest(Context context) { in getInstanceForTest()
72 synchronized (FederatedTrainingTaskDao.class) { in getInstanceForTest()
76 sSingletonInstance = new FederatedTrainingTaskDao(dbHelper); in getInstanceForTest()
/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/scheduling/
DFederatedComputeLearningJobScheduleOrchestrator.java25 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
44 private final FederatedTrainingTaskDao mFederatedTrainingTaskDao;
50 FederatedTrainingTaskDao federatedTrainingTaskDao, in FederatedComputeLearningJobScheduleOrchestrator()
69 FederatedTrainingTaskDao.getInstance(context), in getInstance()
DDeleteExpiredJob.java36 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
75 FederatedTrainingTaskDao getTrainingTaskDao(Context context) { in getTrainingTaskDao()
76 return FederatedTrainingTaskDao.getInstance(context); in getTrainingTaskDao()
DDeleteExpiredJobService.java38 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
78 FederatedTrainingTaskDao getTrainingTaskDao(Context context) { in getTrainingTaskDao()
79 return FederatedTrainingTaskDao.getInstance(context); in getTrainingTaskDao()
DFederatedComputeJobManager.java38 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
65 private final FederatedTrainingTaskDao mFederatedTrainingTaskDao;
74 FederatedTrainingTaskDao federatedTrainingTaskDao, in FederatedComputeJobManager()
97 FederatedTrainingTaskDao.getInstance(mContext), in getInstance()
/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/training/
DEligibilityDecider.java33 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
48 private final FederatedTrainingTaskDao mTaskDao;
53 FederatedTrainingTaskDao taskDao, in EligibilityDecider()
59 public EligibilityDecider(FederatedTrainingTaskDao taskDao) { in EligibilityDecider()
DFederatedComputeWorker.java60 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
1168 return new EligibilityDecider(FederatedTrainingTaskDao.getInstance(context));
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/scheduling/
DFederatedComputeLearningJobScheduleOrchestratorTest.java37 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
59 private FederatedTrainingTaskDao mTrainingTaskDao;
69 mTrainingTaskDao = FederatedTrainingTaskDao.getInstanceForTest(mContext); in setUp()
DDeleteExpiredJobServiceTest.java54 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
90 private FederatedTrainingTaskDao mTrainingTaskDao;
110 mTrainingTaskDao = FederatedTrainingTaskDao.getInstanceForTest(mContext); in setUp()
293 FederatedTrainingTaskDao getTrainingTaskDao(Context context) { in getTrainingTaskDao()
DDeleteExpiredJobTest.java50 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
91 @Mock private FederatedTrainingTaskDao mMockFederatedTrainingTaskDao;
210 FederatedTrainingTaskDao getTrainingTaskDao(Context context) { in getTrainingTaskDao()
DFederatedComputeJobManagerTest.java47 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
122 private FederatedTrainingTaskDao mTrainingTaskDao;
149 mTrainingTaskDao = FederatedTrainingTaskDao.getInstanceForTest(mContext); in setUp()
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/training/
DEligibilityDeciderTest.java49 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
108 private FederatedTrainingTaskDao mTrainingTaskDao;
121 mTrainingTaskDao = FederatedTrainingTaskDao.getInstanceForTest(mContext); in setUp()
DFederatedComputeWorkerTest.java70 import com.android.federatedcompute.services.data.FederatedTrainingTaskDao;
320 private FederatedTrainingTaskDao mTrainingTaskDao;
375 mTrainingTaskDao = FederatedTrainingTaskDao.getInstanceForTest(mContext); in doBeforeEachTest()
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/data/
DFederatedTrainingTaskDaoTest.java79 private FederatedTrainingTaskDao mTrainingTaskDao;
87 mTrainingTaskDao = FederatedTrainingTaskDao.getInstanceForTest(mContext); in setUp()