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()
234 bool EmbeddingNetwork::ComputeLogits(const VectorSpan<float> &input,  in ComputeLogits()
236   return EmbeddingNetwork::ComputeLogitsInternal(input, scores);  in ComputeLogits()
239 bool EmbeddingNetwork::ComputeLogits(const Vector &input,  in ComputeLogits()
241   return EmbeddingNetwork::ComputeLogitsInternal(input, scores);  in ComputeLogits()
244 bool EmbeddingNetwork::ComputeLogitsInternal(const VectorSpan<float> &input,  in ComputeLogitsInternal()
250 bool EmbeddingNetwork::FinishComputeFinalScoresInternal(  in FinishComputeFinalScoresInternal()
281 bool EmbeddingNetwork::ComputeFinalScores(  in ComputeFinalScores()
286 bool EmbeddingNetwork::ComputeFinalScores(  in ComputeFinalScores()
306 EmbeddingNetwork::EmbeddingNetwork(const EmbeddingNetworkParams *model) {  in EmbeddingNetwork()  function in libtextclassifier::nlp_core::EmbeddingNetwork
375 int EmbeddingNetwork::EmbeddingSize(int es_index) const {  in EmbeddingSize()