Home
last modified time | relevance | path

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

/external/libtextclassifier/native/annotator/
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()
Dannotator.cc1316 std::unique_ptr<CachedFeatures> cached_features; in ModelSuggestSelection() local
1324 &cached_features)) { in ModelSuggestSelection()
1332 interpreter_manager->SelectionInterpreter(), *cached_features, in ModelSuggestSelection()
1526 std::unique_ptr<CachedFeatures> cached_features; in ModelClassifyText() local
1532 &cached_features)) { in ModelClassifyText()
1538 features.reserve(cached_features->OutputFeaturesSize()); in ModelClassifyText()
1540 cached_features->AppendBoundsSensitiveFeaturesForSpan(selection_token_span, in ModelClassifyText()
1543 cached_features->AppendClickContextFeaturesForClick(click_pos, &features); in ModelClassifyText()
1998 std::unique_ptr<CachedFeatures> cached_features; in ModelAnnotate() local
2006 &cached_features)) { in ModelAnnotate()
[all …]
Dfeature-processor_test.cc704 std::unique_ptr<CachedFeatures> cached_features; in TEST_F() local
715 &cached_features)); in TEST_F()
717 cached_features->AppendClickContextFeaturesForClick(1, &features); in TEST_F()
746 std::unique_ptr<CachedFeatures> cached_features; in TEST_F() local
769 &cached_features)); in TEST_F()
771 cached_features->AppendBoundsSensitiveFeaturesForSpan({2, 4}, &features); in TEST_F()
Dannotator.h370 const CachedFeatures& cached_features,
378 const CachedFeatures& cached_features,
387 const TokenSpan& inference_span, const CachedFeatures& cached_features,
Dfeature-processor.h174 std::unique_ptr<CachedFeatures>* cached_features) const;
Dfeature-processor.cc795 std::unique_ptr<CachedFeatures>* cached_features) const { in ExtractFeatures()
817 *cached_features = CachedFeatures::Create(token_span, std::move(features), in ExtractFeatures()
820 if (!*cached_features) { in ExtractFeatures()