Home
last modified time | relevance | path

Searched refs:cached_features (Results 1 – 7 of 7) sorted by relevance

/external/libtextclassifier/
Dcached-features_test.cc51 const CachedFeatures& cached_features, int click_pos) { in GetCachedClickContextFeatures() argument
53 cached_features.AppendClickContextFeaturesForClick(click_pos, in GetCachedClickContextFeatures()
59 const CachedFeatures& cached_features, TokenSpan selected_span) { in GetCachedBoundsSensitiveFeatures() argument
61 cached_features.AppendBoundsSensitiveFeaturesForSpan(selected_span, in GetCachedBoundsSensitiveFeatures()
78 const std::unique_ptr<CachedFeatures> cached_features = in TEST() local
83 ASSERT_TRUE(cached_features); in TEST()
85 EXPECT_THAT(GetCachedClickContextFeatures(*cached_features, 5), in TEST()
89 EXPECT_THAT(GetCachedClickContextFeatures(*cached_features, 6), in TEST()
93 EXPECT_THAT(GetCachedClickContextFeatures(*cached_features, 7), in TEST()
119 const std::unique_ptr<CachedFeatures> cached_features = in TEST() local
[all …]
Dcached-features.cc74 std::unique_ptr<CachedFeatures> cached_features(new CachedFeatures()); in Create() local
75 cached_features->extraction_span_ = extraction_span; in Create()
76 cached_features->features_ = std::move(features); in Create()
77 cached_features->padding_features_ = std::move(padding_features); in Create()
78 cached_features->options_ = options; in Create()
80 cached_features->output_features_size_ = in Create()
83 return cached_features; in Create()
Dtext-classifier.cc713 std::unique_ptr<CachedFeatures> cached_features; in ModelSuggestSelection() local
721 &cached_features)) { in ModelSuggestSelection()
729 interpreter_manager->SelectionInterpreter(), *cached_features, in ModelSuggestSelection()
887 std::unique_ptr<CachedFeatures> cached_features; in ModelClassifyText() local
893 &cached_features)) { in ModelClassifyText()
899 features.reserve(cached_features->OutputFeaturesSize()); in ModelClassifyText()
901 cached_features->AppendBoundsSensitiveFeaturesForSpan(selection_token_span, in ModelClassifyText()
904 cached_features->AppendClickContextFeaturesForClick(click_pos, &features); in ModelClassifyText()
1136 std::unique_ptr<CachedFeatures> cached_features; in ModelAnnotate() local
1144 &cached_features)) { in ModelAnnotate()
[all …]
Dtext-classifier.h281 const CachedFeatures& cached_features,
289 const CachedFeatures& cached_features,
298 const TokenSpan& inference_span, const CachedFeatures& cached_features,
Dfeature-processor_test.cc637 std::unique_ptr<CachedFeatures> cached_features; in TEST() local
648 &cached_features)); in TEST()
650 cached_features->AppendClickContextFeaturesForClick(1, &features); in TEST()
680 std::unique_ptr<CachedFeatures> cached_features; in TEST() local
703 &cached_features)); in TEST()
705 cached_features->AppendBoundsSensitiveFeaturesForSpan({2, 4}, &features); in TEST()
Dfeature-processor.h173 std::unique_ptr<CachedFeatures>* cached_features) const;
Dfeature-processor.cc793 std::unique_ptr<CachedFeatures>* cached_features) const { in ExtractFeatures()
815 *cached_features = CachedFeatures::Create(token_span, std::move(features), in ExtractFeatures()
818 if (!*cached_features) { in ExtractFeatures()