Searched refs:embedding_dims (Results 1 - 2 of 2) sorted by relevance

/external/libtextclassifier/common/
H A Dembedding-feature-extractor.cc43 const std::string embedding_dims = local
44 context->Get(GetParamName("embedding_dims"), "");
45 TC_LOG(INFO) << "Embedding dims: " << embedding_dims;
49 for (const std::string &dim : strings::Split(embedding_dims, ';')) {
H A Dembedding-feature-extractor.h75 const std::vector<int> &embedding_dims() const { return embedding_dims_; } function in class:libtextclassifier::nlp_core::GenericEmbeddingFeatureExtractor

Completed in 67 milliseconds