HomeSort by relevance Sort by last modified time
    Searched refs:embedding_matrix (Results 1 - 2 of 2) sorted by null

  /external/libtextclassifier/common/
embedding-network.cc 135 EmbeddingMatrix *const embedding_matrix = local
137 if (embedding_matrix == nullptr) {
149 if (!GetEmbeddingInternal(feature_vector, embedding_matrix, concat_offset,
160 EmbeddingMatrix *const embedding_matrix = embedding_matrices_[es_index].get(); local
161 if (embedding_matrix == nullptr) {
166 return GetEmbeddingInternal(feature_vector, embedding_matrix, 0, embedding,
172 EmbeddingMatrix *const embedding_matrix, const int concat_offset,
174 const int embedding_dim = embedding_matrix->dim();
176 embedding_matrix->quant_type() != QuantizationType::NONE;
204 embedding_matrix->get_embedding(id, &embedding_data, &multiplier)
    [all...]
embedding-network.h 205 EmbeddingMatrix *embedding_matrix,

Completed in 131 milliseconds