Home
last modified time | relevance | path

Searched refs:resumptionToken (Results 1 – 12 of 12) sorted by relevance

/packages/modules/OnDevicePersonalization/tests/frameworktests/src/android/federatedcompute/common/
DExampleConsumptionTest.java80 byte[] resumptionToken = new byte[] {25, 10, 4, 56}; in testBuilder_normalCaseWithResumptionToken()
82 createExampleConsumption(taskId, selectionCriteria, exampleCount, resumptionToken); in testBuilder_normalCaseWithResumptionToken()
88 .isEqualTo(ByteString.copyFrom(resumptionToken)); in testBuilder_normalCaseWithResumptionToken()
96 byte[] resumptionToken = new byte[] {25, 10, 4, 56}; in testWriteToParcel()
98 createExampleConsumption(taskId, selectionCriteria, exampleCount, resumptionToken); in testWriteToParcel()
112 .isEqualTo(ByteString.copyFrom(resumptionToken)); in testWriteToParcel()
119 @Nullable byte[] resumptionToken) { in createExampleConsumption() argument
125 if (resumptionToken != null) { in createExampleConsumption()
126 builder.setResumptionToken(resumptionToken); in createExampleConsumption()
/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/examplestore/
DFederatedExampleIterator.java96 byte[] resumptionToken, in FederatedExampleIterator() argument
100 this.mResumptionToken = resumptionToken; in FederatedExampleIterator()
177 private IteratorResult(byte[] resultBytes, byte[] resumptionToken) { in IteratorResult() argument
179 this.mResumptionToken = resumptionToken; in IteratorResult()
298 byte[] resumptionToken = result.getByteArray(EXTRA_EXAMPLE_ITERATOR_RESUMPTION_TOKEN); in onIteratorNextSuccess()
299 if (resumptionToken == null) { in onIteratorNextSuccess()
300 resumptionToken = new byte[] {}; in onIteratorNextSuccess()
303 Pair.create(new IteratorResult(example, resumptionToken), null)); in onIteratorNextSuccess()
DExampleConsumptionRecorder.java68 @Nullable byte[] resumptionToken) { in incrementAndUpdateResumptionToken()
71 resumptionToken == null in incrementAndUpdateResumptionToken()
73 : Arrays.copyOf(resumptionToken, resumptionToken.length); in incrementAndUpdateResumptionToken()
DExampleStoreServiceProvider.java82 byte[] resumptionToken = exampleSelector.getResumptionToken().toByteArray(); in getExampleIterator()
84 ClientConstants.EXTRA_EXAMPLE_ITERATOR_RESUMPTION_TOKEN, resumptionToken); in getExampleIterator()
/packages/modules/OnDevicePersonalization/framework/java/android/adservices/ondevicepersonalization/
DTrainingExampleRecord.java66 @Nullable byte[] resumptionToken) { in TrainingExampleRecord() argument
68 this.mResumptionToken = resumptionToken; in TrainingExampleRecord()
115 byte[] resumptionToken = in.createByteArray(); in TrainingExampleRecord()
118 this.mResumptionToken = resumptionToken; in TrainingExampleRecord()
DTrainingExamplesInputParcel.java65 @Nullable byte[] resumptionToken, in TrainingExamplesInputParcel() argument
71 this.mResumptionToken = resumptionToken; in TrainingExamplesInputParcel()
137 byte[] resumptionToken = in.createByteArray(); in TrainingExamplesInputParcel()
145 this.mResumptionToken = resumptionToken; in TrainingExamplesInputParcel()
DTrainingExamplesInput.java99 @Nullable byte[] resumptionToken, in TrainingExamplesInput() argument
105 this.mResumptionToken = resumptionToken; in TrainingExamplesInput()
/packages/modules/OnDevicePersonalization/federatedcompute/src/com/android/federatedcompute/services/training/
DComputationRunner.java53 byte[] resumptionToken = exampleSelector.getResumptionToken().toByteArray(); in runTaskWithNativeRunner()
57 resumptionToken, in runTaskWithNativeRunner()
58 recorder.createRecorderForTracking(taskName, resumptionToken), in runTaskWithNativeRunner()
/packages/modules/OnDevicePersonalization/framework/java/android/federatedcompute/common/
DExampleConsumption.java63 @Nullable byte[] resumptionToken, in ExampleConsumption() argument
69 this.mResumptionToken = resumptionToken; in ExampleConsumption()
168 byte[] resumptionToken = in.createByteArray(); in ExampleConsumption()
175 this.mResumptionToken = resumptionToken; in ExampleConsumption()
/packages/modules/OnDevicePersonalization/src/com/android/ondevicepersonalization/services/federatedcompute/
DOdpResultHandlingService.java117 byte[] resumptionToken = consumption.getResumptionToken(); in processExampleConsumptions()
118 if (resumptionToken != null) { in processExampleConsumptions()
123 .setToken(resumptionToken) in processExampleConsumptions()
DOdpExampleStoreIterator.java45 byte[] resumptionToken = record.getResumptionToken(); in next()
48 result.putByteArray(EXTRA_EXAMPLE_ITERATOR_RESUMPTION_TOKEN, resumptionToken); in next()
DOdpExampleStoreService.java174 byte[] resumptionToken = null; in startQuery()
176 resumptionToken = eventState.getToken(); in startQuery()
181 .setResumptionToken(resumptionToken) in startQuery()