Home
last modified time | relevance | path

Searched refs:InferenceInput (Results 1 – 10 of 10) sorted by relevance

/packages/modules/OnDevicePersonalization/tests/cts/endtoend/src/com/android/ondevicepersonalization/cts/e2e/
DInferenceInputTest.java25 import android.adservices.ondevicepersonalization.InferenceInput;
54 InferenceInput.Params params = in buildInput_success()
55 new InferenceInput.Params.Builder(mRemoteData, MODEL_KEY).build(); in buildInput_success()
57 InferenceInput inferenceInput = in buildInput_success()
58 new InferenceInput.Builder( in buildInput_success()
78 InferenceInput.Params params = in buildInputWithSetters()
79 new InferenceInput.Params.Builder(mRemoteData, MODEL_KEY).build(); in buildInputWithSetters()
81 InferenceInput inferenceInput = in buildInputWithSetters()
82 new InferenceInput.Builder( in buildInputWithSetters()
106 InferenceInput.Params params = in buildInput_batchNotSet_success()
[all …]
/packages/modules/OnDevicePersonalization/tests/frameworktests/src/android/adservices/ondevicepersonalization/
DInferenceInputTest.java47 InferenceInput.Params.Builder builder = in buildParams_reusable()
48 new InferenceInput.Params.Builder(mRemoteData, MODEL_KEY); in buildParams_reusable()
51 InferenceInput.Params params = builder.setModelKey("other_kay").build(); in buildParams_reusable()
62 InferenceInput.Params params = in buildInferenceInput_reusable()
63 new InferenceInput.Params.Builder(mRemoteData, MODEL_KEY).build(); in buildInferenceInput_reusable()
65 InferenceInput.Builder builder = in buildInferenceInput_reusable()
66 new InferenceInput.Builder( in buildInferenceInput_reusable()
72 InferenceInput inferenceInput = builder.setBatchSize(10).build(); in buildInferenceInput_reusable()
82 InferenceInput.Params params = in buildInput_success()
83 new InferenceInput.Params.Builder(mRemoteData, MODEL_KEY).build(); in buildInput_success()
[all …]
DModelManagerTest.java72 InferenceInput inferenceContext = in runInference_success()
73 new InferenceInput.Builder( in runInference_success()
74 new InferenceInput.Params.Builder(mRemoteData, MODEL_KEY).build(), in runInference_success()
95 InferenceInput inferenceContext = in runInference_error()
96 new InferenceInput.Builder( in runInference_error()
97 new InferenceInput.Params.Builder(mRemoteData, INVALID_MODEL_KEY) in runInference_error()
125 InferenceInput inferenceContext = in runInference_resultMissingInferenceOutput()
126 new InferenceInput.Builder( in runInference_resultMissingInferenceOutput()
127 new InferenceInput.Params.Builder(mRemoteData, MISSING_OUTPUT_KEY) in runInference_resultMissingInferenceOutput()
/packages/modules/OnDevicePersonalization/tests/servicetests/src/com/android/ondevicepersonalization/services/inference/
DIsolatedModelServiceImplTest.java28 import android.adservices.ondevicepersonalization.InferenceInput;
65 private InferenceInput.Params mParams;
73 mParams = new InferenceInput.Params.Builder(mRemoteData, MODEL_KEY).build(); in setup()
79 InferenceInput inferenceInput = in runModelInference_singleExample_success()
80 new InferenceInput.Builder( in runModelInference_singleExample_success()
106 InferenceInput inferenceInput = in runModelInference_setBatchSizeNotMatch_success()
107 new InferenceInput.Builder( in runModelInference_setBatchSizeNotMatch_success()
135 InferenceInput inferenceInput = in runModelInferenceBatch_setBatchSize_success()
136 new InferenceInput.Builder( in runModelInferenceBatch_setBatchSize_success()
164 InferenceInput inferenceInput = in runModelInferenceBatch_notSetBatchSize_success()
[all …]
/packages/modules/OnDevicePersonalization/framework/java/android/adservices/ondevicepersonalization/
DInferenceInputParcel.java40 private @InferenceInput.Params.Delegate int mDelegate;
59 private @InferenceInput.Params.ModelType int mModelType =
60 InferenceInput.Params.MODEL_TYPE_TENSORFLOW_LITE;
70 public InferenceInputParcel(@NonNull InferenceInput value) { in InferenceInputParcel()
120 @InferenceInput.Params.Delegate int delegate, in InferenceInputParcel()
124 @InferenceInput.Params.ModelType int modelType, in InferenceInputParcel()
129 AnnotationValidations.validate(InferenceInput.Params.Delegate.class, null, mDelegate); in InferenceInputParcel()
136 AnnotationValidations.validate(InferenceInput.Params.ModelType.class, null, mModelType); in InferenceInputParcel()
153 public @InferenceInput.Params.Delegate int getDelegate() { in getDelegate()
184 public @InferenceInput.Params.ModelType int getModelType() { in getModelType()
[all …]
DInferenceInput.java38 public final class InferenceInput { class
411 /* package-private */ InferenceInput( in InferenceInput() method in InferenceInput
491 InferenceInput that = (InferenceInput) o; in equals()
628 public @NonNull InferenceInput build() { in build()
634 InferenceInput o = in build()
635 new InferenceInput(mParams, mInputData, mBatchSize, mExpectedOutputStructure); in build()
DModelManager.java65 @NonNull InferenceInput input, in run()
/packages/modules/OnDevicePersonalization/framework/api/
Dcurrent.txt108 …epersonalization.flags.on_device_personalization_apis_enabled") public final class InferenceInput {
112 …method @NonNull public android.adservices.ondevicepersonalization.InferenceInput.Params getParams(…
115 public static final class InferenceInput.Builder {
116 …ctor public InferenceInput.Builder(@NonNull android.adservices.ondevicepersonalization.InferenceIn…
117 method @NonNull public android.adservices.ondevicepersonalization.InferenceInput build();
118 …method @NonNull public android.adservices.ondevicepersonalization.InferenceInput.Builder setBatchS…
119 …method @NonNull public android.adservices.ondevicepersonalization.InferenceInput.Builder setExpect…
120 …method @NonNull public android.adservices.ondevicepersonalization.InferenceInput.Builder setInputD…
121 …es.ondevicepersonalization.InferenceInput.Builder setParams(@NonNull android.adservices.ondevicepe…
124 public static class InferenceInput.Params {
[all …]
/packages/modules/OnDevicePersonalization/tests/cts/service/src/com/android/ondevicepersonalization/testing/sampleservice/
DSampleWorker.java25 import android.adservices.ondevicepersonalization.InferenceInput;
300 InferenceInput.Params params = in handleRunModelInference()
301 new InferenceInput.Params.Builder(mLocalData, tableKey).build(); in handleRunModelInference()
302 InferenceInput input = buildInferenceInput(params); in handleRunModelInference()
333 private InferenceInput buildInferenceInput(InferenceInput.Params params) { in buildInferenceInput()
341 InferenceInput input = in buildInferenceInput()
342 new InferenceInput.Builder( in buildInferenceInput()
/packages/modules/OnDevicePersonalization/samples/odpsamplenetwork/src/main/java/com/example/odpsamplenetwork/
DSampleHandler.java30 import android.adservices.ondevicepersonalization.InferenceInput;
907 InferenceInput.Params params =
908 new InferenceInput.Params.Builder(mRemoteData, "model1").build();
909 InferenceInput input =
910 new InferenceInput.Builder(