Home
last modified time | relevance | path

Searched refs:EmbeddingVector (Results 1 – 11 of 11) sorted by relevance

/packages/modules/AppSearch/framework/java/external/android/app/appsearch/
DEmbeddingVector.java48 public final class EmbeddingVector extends AbstractSafeParcelable { class
51 public static final Parcelable.Creator<EmbeddingVector> CREATOR = new EmbeddingVectorCreator();
69 public EmbeddingVector( in EmbeddingVector() method in EmbeddingVector
97 if (!(o instanceof EmbeddingVector)) return false; in equals()
98 EmbeddingVector that = (EmbeddingVector) o; in equals()
DGenericDocument.java373 EmbeddingVector[] embeddingValues = propertyParcel.getEmbeddingValues(); in getRawPropertyFromRawDocument()
686 public EmbeddingVector getPropertyEmbedding(@NonNull String path) { in getPropertyEmbedding()
688 EmbeddingVector[] propertyArray = getPropertyEmbeddingArray(path); in getPropertyEmbedding()
871 public EmbeddingVector[] getPropertyEmbeddingArray(@NonNull String path) { in getPropertyEmbeddingArray()
874 return safeCastProperty(path, value, EmbeddingVector[].class); in getPropertyEmbeddingArray()
1371 @NonNull String name, @NonNull EmbeddingVector... values) { in setPropertyEmbedding()
DSearchSpec.java143 private final List<EmbeddingVector> mSearchEmbeddings;
359 @Param(id = 20) @Nullable List<EmbeddingVector> searchEmbeddings, in SearchSpec()
666 public List<EmbeddingVector> getSearchEmbeddings() { in getSearchEmbeddings()
751 private List<EmbeddingVector> mSearchEmbeddings = new ArrayList<>();
1571 public Builder addSearchEmbeddings(@NonNull EmbeddingVector... searchEmbeddings) { in addSearchEmbeddings()
1587 public Builder addSearchEmbeddings(@NonNull Collection<EmbeddingVector> searchEmbeddings) { in addSearchEmbeddings()
/packages/modules/AppSearch/service/java/com/android/server/appsearch/external/localstorage/converter/
DGenericDocumentToProtoConverter.java21 import android.app.appsearch.EmbeddingVector;
57 private static final EmbeddingVector[] EMPTY_EMBEDDING_ARRAY = new EmbeddingVector[0];
111 } else if (property instanceof EmbeddingVector[]) { in toDocumentProto()
112 EmbeddingVector[] embeddingValues = (EmbeddingVector[]) property; in toDocumentProto()
216 EmbeddingVector[] values = new EmbeddingVector[property.getVectorValuesCount()]; in toGenericDocument()
233 public static EmbeddingVector vectorProtoToEmbeddingVector( in vectorProtoToEmbeddingVector()
241 return new EmbeddingVector(values, vectorProto.getModelSignature()); in vectorProtoToEmbeddingVector()
247 @NonNull EmbeddingVector embedding) { in embeddingVectorToVectorProto()
DSearchSpecToProtoConverter.java26 import android.app.appsearch.EmbeddingVector;
292 List<EmbeddingVector> searchEmbeddings = mSearchSpec.getSearchEmbeddings(); in toSearchSpecProto()
/packages/modules/AppSearch/framework/java/external/android/app/appsearch/safeparcel/
DPropertyParcel.java22 import android.app.appsearch.EmbeddingVector;
74 private final EmbeddingVector[] mEmbeddingValues;
87 @Param(id = 8) @Nullable EmbeddingVector[] embeddingValues) { in PropertyParcel()
143 public EmbeddingVector[] getEmbeddingValues() { in getEmbeddingValues()
271 private EmbeddingVector[] mEmbeddingValues;
328 public Builder setEmbeddingValues(@NonNull EmbeddingVector[] embeddingValues) { in setEmbeddingValues()
DGenericDocumentParcel.java25 import android.app.appsearch.EmbeddingVector;
511 public Builder putInPropertyMap(@NonNull String name, @NonNull EmbeddingVector[] values) { in putInPropertyMap()
/packages/modules/AppSearch/testing/coretests/src/android/app/appsearch/external/app/
DSearchSpecInternalTest.java126 EmbeddingVector embedding1 = in testSearchSpecBuilderCopyConstructor_embeddingSearch()
127 new EmbeddingVector(new float[] {1.1f, 2.2f, 3.3f}, "my_model_v1"); in testSearchSpecBuilderCopyConstructor_embeddingSearch()
128 EmbeddingVector embedding2 = in testSearchSpecBuilderCopyConstructor_embeddingSearch()
129 new EmbeddingVector(new float[] {4.4f, 5.5f, 6.6f, 7.7f}, "my_model_v2"); in testSearchSpecBuilderCopyConstructor_embeddingSearch()
/packages/modules/AppSearch/testing/servicestests/src/com/android/server/appsearch/external/localstorage/converter/
DGenericDocumentToProtoConverterTest.java21 import android.app.appsearch.EmbeddingVector;
50 private static final EmbeddingVector sEmbedding1 =
51 new EmbeddingVector(new float[] {1.1f, 2.2f, 3.3f}, "my_model_v1");
52 private static final EmbeddingVector sEmbedding2 =
53 new EmbeddingVector(new float[] {4.4f, 5.5f, 6.6f, 7.7f}, "my_model_v2");
/packages/modules/AppSearch/testing/coretests/src/android/app/appsearch/external/safeparcel/
DGenericDocumentParcelTest.java23 import android.app.appsearch.EmbeddingVector;
42 EmbeddingVector[] embeddingValues = {new EmbeddingVector(new float[1], "my_model")}; in testPropertyParcel_onePropertySet_success()
/packages/modules/AppSearch/framework/api/
Dcurrent.txt204 …h.flags.enable_schema_embedding_property_config") public final class EmbeddingVector implements an…
205 ctor public EmbeddingVector(@NonNull float[], @NonNull String);
210 …l public static final android.os.Parcelable.Creator<android.app.appsearch.EmbeddingVector> CREATOR;
234 …a_embedding_property_config") @Nullable public android.app.appsearch.EmbeddingVector getPropertyEm…
235 …a_embedding_property_config") @Nullable public android.app.appsearch.EmbeddingVector[] getProperty…
258 …ilderType setPropertyEmbedding(@NonNull String, @NonNull android.app.appsearch.EmbeddingVector...);
529 …operty_config") @NonNull public java.util.List<android.app.appsearch.EmbeddingVector> getSearchEmb…
580 …ppsearch.SearchSpec.Builder addSearchEmbeddings(@NonNull android.app.appsearch.EmbeddingVector...);
581 ….Builder addSearchEmbeddings(@NonNull java.util.Collection<android.app.appsearch.EmbeddingVector>);