Home
last modified time | relevance | path

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

/external/libtextclassifier/common/
Dembedding-network.cc45 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 …]
Dembedding-network.h46 class EmbeddingNetwork {
156 explicit EmbeddingNetwork(const EmbeddingNetworkParams *model);
158 virtual ~EmbeddingNetwork() {} in ~EmbeddingNetwork()
Dembedding-network.proto15 // 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/
Dembedding-network_test.cc31 class TestingEmbeddingNetwork : public EmbeddingNetwork {
33 using EmbeddingNetwork::EmbeddingNetwork;
34 using EmbeddingNetwork::FinishComputeFinalScoresInternal;
/external/libtextclassifier/smartselect/
Dtext-classification-model.h95 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_;
Dtext-classification-model.cc36 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/
Dlang-id.cc139 network_.reset(new EmbeddingNetwork(network_params_.get())); in Initialize()
217 EmbeddingNetwork::Vector scores; in ScoreLanguages()
323 std::unique_ptr<EmbeddingNetwork> network_;