Searched refs:EmbeddingNetwork (Results 1 – 7 of 7) sorted by relevance
/external/libtextclassifier/common/ |
D | embedding-network.cc | 45 EmbeddingNetwork::Matrix *mat) { in InitNonQuantizedMatrix() 54 (*mat)[r] = EmbeddingNetwork::VectorWrapper(weights, source_matrix.cols); in InitNonQuantizedMatrix() 66 EmbeddingNetwork::VectorWrapper *vector) { in InitNonQuantizedVector() 80 *vector = EmbeddingNetwork::VectorWrapper(weights, source_matrix.rows); in InitNonQuantizedVector() 87 const EmbeddingNetwork::Matrix &weights, in SparseReluProductPlusBias() 88 const EmbeddingNetwork::VectorWrapper &b, in SparseReluProductPlusBias() 90 EmbeddingNetwork::Vector *y) { in SparseReluProductPlusBias() 119 bool EmbeddingNetwork::ConcatEmbeddings( in ConcatEmbeddings() 158 bool EmbeddingNetwork::GetEmbedding(const FeatureVector &feature_vector, in GetEmbedding() 170 bool EmbeddingNetwork::GetEmbeddingInternal( in GetEmbeddingInternal() [all …]
|
D | embedding-network.h | 46 class EmbeddingNetwork { 156 explicit EmbeddingNetwork(const EmbeddingNetworkParams *model); 158 virtual ~EmbeddingNetwork() {} in ~EmbeddingNetwork()
|
D | embedding-network.proto | 15 // Protos for performing inference with an EmbeddingNetwork. 49 // Stores all parameters for a given EmbeddingNetwork. This can either be a 50 // EmbeddingNetwork or a PrecomputedEmbeddingNetwork: for precomputed networks,
|
/external/libtextclassifier/tests/ |
D | embedding-network_test.cc | 31 class TestingEmbeddingNetwork : public EmbeddingNetwork { 33 using EmbeddingNetwork::EmbeddingNetwork; 34 using EmbeddingNetwork::FinishComputeFinalScoresInternal;
|
/external/libtextclassifier/smartselect/ |
D | text-classification-model.h | 95 nlp_core::EmbeddingNetwork::Vector InferInternal( 98 const nlp_core::EmbeddingNetwork& network, 115 std::unique_ptr<nlp_core::EmbeddingNetwork> selection_network_; 119 std::unique_ptr<nlp_core::EmbeddingNetwork> sharing_network_;
|
D | text-classification-model.cc | 36 using nlp_core::EmbeddingNetwork; 133 FeatureVectorFn CreateFeatureVectorFn(const EmbeddingNetwork& network, in CreateFeatureVectorFn() 188 selection_network_.reset(new EmbeddingNetwork(selection_params_.get())); in LoadModels() 200 sharing_network_.reset(new EmbeddingNetwork(sharing_params_.get())); in LoadModels() 234 EmbeddingNetwork::Vector TextClassificationModel::InferInternal( in InferInternal() 236 const FeatureProcessor& feature_processor, const EmbeddingNetwork& network, in InferInternal() 347 EmbeddingNetwork::Vector scores = InferInternal( in SuggestSelectionInternal() 478 EmbeddingNetwork::Vector scores = in ClassifyText()
|
/external/libtextclassifier/lang_id/ |
D | lang-id.cc | 139 network_.reset(new EmbeddingNetwork(network_params_.get())); in Initialize() 217 EmbeddingNetwork::Vector scores; in ScoreLanguages() 323 std::unique_ptr<EmbeddingNetwork> network_;
|