Searched defs:feature_vector_fn (Results 1 - 3 of 3) sorted by relevance

/external/libtextclassifier/smartselect/
H A Dcached-features.cc26 float*)>& feature_vector_fn) {
29 feature_vector_fn(sparse_features[i], dense_features[i],
22 Extract( const std::vector<std::vector<int>>& sparse_features, const std::vector<std::vector<float>>& dense_features, const std::function<bool(const std::vector<int>&, const std::vector<float>&, float*)>& feature_vector_fn) argument
H A Dcached-features.h37 // - feature_vector_fn: Writes features for given Token to the specified
47 feature_vector_fn,
54 Extract(sparse_features, dense_features, feature_vector_fn);
79 feature_vector_fn);
42 CachedFeatures(VectorSpan<Token> tokens, int context_size, const std::vector<std::vector<int>>& sparse_features, const std::vector<std::vector<float>>& dense_features, const std::function<bool(const std::vector<int>&, const std::vector<float>&, float*)>& feature_vector_fn, int feature_vector_size) argument
H A Dtext-classification-model.cc237 const FeatureVectorFn& feature_vector_fn,
234 InferInternal( const std::string& context, CodepointSpan span, const FeatureProcessor& feature_processor, const EmbeddingNetwork& network, const FeatureVectorFn& feature_vector_fn, std::vector<CodepointSpan>* selection_label_spans) const argument

Completed in 256 milliseconds