Searched refs:embedding_model (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/out/soong/.intermediates/external/libtextclassifier/native/libtextclassifier_fbgen_actions_actions_model/gen/actions/ |
D | actions_model_generated.h | 807 std::vector<uint8_t> embedding_model{}; 858 const flatbuffers::Vector<uint8_t> *embedding_model() const { in embedding_model() function 917 verifier.VerifyVector(embedding_model()) && in Verify() 952 void add_embedding_model(flatbuffers::Offset<flatbuffers::Vector<uint8_t>> embedding_model) { in add_embedding_model() 953 fbb_.AddOffset(ActionsTokenFeatureProcessorOptions::VT_EMBEDDING_MODEL, embedding_model); in add_embedding_model() 1020 flatbuffers::Offset<flatbuffers::Vector<uint8_t>> embedding_model = 0, 1052 builder_.add_embedding_model(embedding_model); 1064 const std::vector<uint8_t> *embedding_model = nullptr, 1082 if (embedding_model) { _fbb.ForceVectorAlignment(embedding_model->size(), sizeof(uint8_t), 16); } 1083 auto embedding_model__ = embedding_model ? _fbb.CreateVector<uint8_t>(*embedding_model) : 0; [all …]
|
/aosp_15_r20/out/soong/.intermediates/external/libtextclassifier/native/libtextclassifier_fbgen_annotator_model/gen/annotator/ |
D | model_generated.h | 3223 std::vector<uint8_t> embedding_model{}; 3314 const flatbuffers::Vector<uint8_t> *embedding_model() const { in embedding_model() function 3408 verifier.VerifyVector(embedding_model()) && in Verify() 3489 void add_embedding_model(flatbuffers::Offset<flatbuffers::Vector<uint8_t>> embedding_model) { in add_embedding_model() 3490 fbb_.AddOffset(Model::VT_EMBEDDING_MODEL, embedding_model); in add_embedding_model() 3587 flatbuffers::Offset<flatbuffers::Vector<uint8_t>> embedding_model = 0, 3638 builder_.add_embedding_model(embedding_model); 3659 const std::vector<uint8_t> *embedding_model = nullptr, 3691 if (embedding_model) { _fbb.ForceVectorAlignment(embedding_model->size(), sizeof(uint8_t), 16); } 3692 auto embedding_model__ = embedding_model ? _fbb.CreateVector<uint8_t>(*embedding_model) : 0; [all …]
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_expanded_weights.py | 590 def embedding_model(num_classes, num_embedding): function 597 return self._test_embedding_model(embedding_model, 16, device)
|
/aosp_15_r20/external/libtextclassifier/native/actions/ |
H A D | actions_model.fbs | 164 embedding_model:[ubyte] (force_align: 16);
|
H A D | actions-suggestions.cc | 378 options->embedding_model(), options->embedding_size(), in ValidateAndInitialize()
|
/aosp_15_r20/external/libtextclassifier/native/annotator/ |
H A D | annotator.cc | 391 if (!model_->embedding_model()) { in ValidateAndInitialize() 409 model_->embedding_model(), in ValidateAndInitialize()
|
H A D | model.fbs | 565 embedding_model:[ubyte] (force_align: 16);
|