Home
last modified time | relevance | path

Searched refs:GenericEmbeddingFeatureExtractor (Results 1 – 2 of 2) sorted by relevance

/external/libtextclassifier/native/lang_id/common/
Dembedding-feature-extractor.h43 class GenericEmbeddingFeatureExtractor {
49 explicit GenericEmbeddingFeatureExtractor(const std::string &arg_prefix) in GenericEmbeddingFeatureExtractor() function
52 virtual ~GenericEmbeddingFeatureExtractor() {} in ~GenericEmbeddingFeatureExtractor()
105 class EmbeddingFeatureExtractor : public GenericEmbeddingFeatureExtractor {
112 : GenericEmbeddingFeatureExtractor(arg_prefix) {} in EmbeddingFeatureExtractor()
116 if (!GenericEmbeddingFeatureExtractor::Setup(context)) { in Setup()
130 if (!GenericEmbeddingFeatureExtractor::Init(context)) return false; in Init()
Dembedding-feature-extractor.cc36 bool GenericEmbeddingFeatureExtractor::Setup(TaskContext *context) { in Setup()
68 bool GenericEmbeddingFeatureExtractor::Init(TaskContext *context) { in Init()