Searched defs:feature_vector_fn (Results 1 - 3 of 3) sorted by relevance
/external/libtextclassifier/smartselect/ |
H A D | cached-features.cc | 26 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 D | cached-features.h | 37 // - 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 D | text-classification-model.cc | 237 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