Home
last modified time | relevance | path

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

/packages/modules/AdServices/adservices/service-core/java/com/android/adservices/service/topics/classifier/
DOnDeviceClassifier.java74 private final ModelManager mModelManager; field in OnDeviceClassifier
96 mModelManager = modelManager; in OnDeviceClassifier()
109 if (!mModelManager.isModelAvailable()) { in classify()
317 mLabels = mModelManager.retrieveLabels(); in load()
320 mClassifierInputConfig = mModelManager.retrieveClassifierInputConfig(); in load()
330 mModelManager.retrieveClassifierAssetsMetadata(); in load()
337 mBuildId = Ints.saturatedCast(mModelManager.getBuildId()); in load()
343 return BertNLClassifier.createFromBuffer(mModelManager.retrieveModel()); in loadModel()
DPrecomputedClassifier.java65 private final ModelManager mModelManager; field in PrecomputedClassifier
83 mModelManager = modelManager; in PrecomputedClassifier()
184 mLabels = mModelManager.retrieveLabels(); in load()
185 mAppTopics = mModelManager.retrieveAppClassificationTopics(); in load()
195 mModelManager.retrieveClassifierAssetsMetadata(); in load()
202 mBuildId = Ints.saturatedCast(mModelManager.getBuildId()); in load()
/packages/modules/OnDevicePersonalization/tests/frameworktests/src/android/adservices/ondevicepersonalization/
DModelManagerTest.java48 ModelManager mModelManager = field in ModelManagerTest
80 mModelManager.run(inferenceContext, MoreExecutors.directExecutor(), callback); in runInference_success()
104 mModelManager.run(inferenceContext, MoreExecutors.directExecutor(), callback); in runInference_error()
115 mModelManager.run( in runInference_contextNull_throw()
134 mModelManager.run(inferenceContext, MoreExecutors.directExecutor(), callback); in runInference_resultMissingInferenceOutput()
/packages/modules/AdServices/adservices/tests/unittest/service-core/src/com/android/adservices/service/topics/classifier/
DPrecomputedClassifierTest.java72 private ModelManager mModelManager; field in PrecomputedClassifierTest
96 mModelManager = in setUp()
106 sPrecomputedClassifier = new PrecomputedClassifier(mModelManager, mCacheManager, mLogger); in setUp()
199 assertThat(mModelManager.retrieveAppClassificationTopics().get(sampleAppPackageName)) in testClassify_appWithBlockedTopic()
222 assertThat(mModelManager.retrieveAppClassificationTopics().get(sampleAppPackageName)) in testClassify_appWithAllTopicsBlocked()
DOnDeviceClassifierTest.java83 @Mock private ModelManager mModelManager; field in OnDeviceClassifierTest
111 mModelManager = in setUp()
125 mModelManager, in setUp()
136 mModelManager = in testClassify_earlyReturnIfNoModelAvailable()
149 mModelManager, in testClassify_earlyReturnIfNoModelAvailable()
321 mModelManager, in testClassify_appWithBlockedTopic()
362 mModelManager, in testClassify_appWithAllTopicsBlocked()
/packages/modules/OnDevicePersonalization/tests/cts/service/src/com/android/ondevicepersonalization/testing/sampleservice/
DSampleWorker.java74 private final ModelManager mModelManager; field in SampleWorker
92 mModelManager = modelManager; in SampleWorker()
320 mModelManager.run(input, mExecutor, callback); in handleRunModelInference()
/packages/modules/OnDevicePersonalization/samples/odpsamplenetwork/src/main/java/com/example/odpsamplenetwork/
DSampleHandler.java129 private final ModelManager mModelManager; field in SampleHandler
143 mModelManager = modelManager; in SampleHandler()
159 if (mModelManager == null) { in SampleHandler()
916 mModelManager.run(