Searched refs:GenericEmbeddingFeatureExtractor (Results 1 – 2 of 2) sorted by relevance
45 class GenericEmbeddingFeatureExtractor {51 explicit GenericEmbeddingFeatureExtractor(absl::string_view arg_prefix) in GenericEmbeddingFeatureExtractor() function54 virtual ~GenericEmbeddingFeatureExtractor() {} in ~GenericEmbeddingFeatureExtractor()104 class EmbeddingFeatureExtractor : public GenericEmbeddingFeatureExtractor {111 : GenericEmbeddingFeatureExtractor(arg_prefix) {} in EmbeddingFeatureExtractor()115 if (!GenericEmbeddingFeatureExtractor::Setup(context)) { in Setup()129 if (!GenericEmbeddingFeatureExtractor::Init(context)) return false; in Init()
36 bool GenericEmbeddingFeatureExtractor::Setup(TaskContext *context) { in Setup()68 bool GenericEmbeddingFeatureExtractor::Init(TaskContext *context) { in Init()