Home
last modified time | relevance | path

Searched refs:mExampleIterator (Results 1 – 7 of 7) sorted by relevance

/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/training/jni/
DJavaExampleStore.java42 private final ExampleIterator mExampleIterator; field in JavaExampleStore
49 this.mExampleIterator = exampleIterator; in JavaExampleStore()
68 mServedExampleIterators.add(this.mExampleIterator); in createExampleIteratorWithContext()
76 mIterator = mExampleIterator; in createExampleIteratorWithContext()
108 iteratorOpen = mServedExampleIterators.remove(mExampleIterator); in createExampleIteratorWithContext()
DFlRunnerWrapper.java46 private final ExampleIterator mExampleIterator; field in FlRunnerWrapper
54 this.mExampleIterator = exampleIterator; in FlRunnerWrapper()
66 new SimpleTaskEnvironmentImpl(mInterruptionFlag, mExampleIterator); in run()
/packages/modules/OnDevicePersonalization/src/com/android/ondevicepersonalization/services/federatedcompute/
DOdpExampleStoreIterator.java34 ListIterator<TrainingExampleRecord> mExampleIterator; field in OdpExampleStoreIterator
37 mExampleIterator = exampleRecordList.listIterator(); in OdpExampleStoreIterator()
42 if (mExampleIterator.hasNext()) { in next()
43 TrainingExampleRecord record = mExampleIterator.next(); in next()
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/examplestore/
DSampleExampleStoreService.java78 private final Iterator<Example> mExampleIterator; field in SampleExampleStoreService.ListExampleStoreIterator
81 mExampleIterator = examples.iterator(); in ListExampleStoreIterator()
86 if (mExampleIterator.hasNext()) { in next()
89 EXTRA_EXAMPLE_ITERATOR_RESULT, mExampleIterator.next().toByteArray()); in next()
/packages/modules/OnDevicePersonalization/tests/federatedcomputetests/src/com/android/federatedcompute/services/training/jni/
DSimpleTaskEnvironmentImplTest.java40 @Mock ExampleIterator mExampleIterator; field in SimpleTaskEnvironmentImplTest
46 when(mExampleIterator.hasNext()).thenReturn(false); in setUp()
47 mNativeRunnerDeps = new SimpleTaskEnvironmentImpl(mInterruptionFlag, mExampleIterator); in setUp()
DFlRunnerWrapperTest.java82 @Mock ExampleIterator mExampleIterator; field in FlRunnerWrapperTest
103 when(mExampleIterator.hasNext()).thenReturn(false); in testRunInvalidPlan_returnFail()
106 new FlRunnerWrapper(mInterruptionFlag, POPULATION_NAME, mExampleIterator); in testRunInvalidPlan_returnFail()
/packages/modules/OnDevicePersonalization/tests/frameworktests/src/android/federatedcompute/
DExampleStoreServiceTest.java127 private final Iterator<Example> mExampleIterator; field in ExampleStoreServiceTest.ListJavaExampleStoreIterator
130 mExampleIterator = examples.iterator(); in ListJavaExampleStoreIterator()