Home
last modified time | relevance | path

Searched refs:output_features (Results 1 – 9 of 9) sorted by relevance

/external/libtextclassifier/native/annotator/
Dcached-features.cc87 int click_pos, std::vector<float>* output_features) const { in AppendClickContextFeaturesForClick()
93 /*read_mask_span=*/{0, extraction_span_.Size()}, output_features); in AppendClickContextFeaturesForClick()
97 TokenSpan selected_span, std::vector<float>* output_features) const { in AppendBoundsSensitiveFeaturesForSpan()
111 /*read_mask_span=*/{0, selected_span.second}, output_features); in AppendBoundsSensitiveFeaturesForSpan()
121 {selected_span.first, extraction_span_.Size()}, output_features); in AppendBoundsSensitiveFeaturesForSpan()
124 AppendBagFeatures(selected_span, output_features); in AppendBoundsSensitiveFeaturesForSpan()
128 output_features->push_back(static_cast<float>(selected_span.Size())); in AppendBoundsSensitiveFeaturesForSpan()
134 std::vector<float>* output_features) const { in AppendFeaturesInternal()
138 AppendPaddingFeatures(output_features); in AppendFeaturesInternal()
140 output_features->insert( in AppendFeaturesInternal()
[all …]
Dcached-features.h42 int click_pos, std::vector<float>* output_features) const;
47 TokenSpan selected_span, std::vector<float>* output_features) const;
61 std::vector<float>* output_features) const;
64 void AppendPaddingFeatures(std::vector<float>* output_features) const;
70 std::vector<float>* output_features) const;
Dcached-features_test.cc52 std::vector<float> output_features; in GetCachedClickContextFeatures() local
54 &output_features); in GetCachedClickContextFeatures()
55 return output_features; in GetCachedClickContextFeatures()
60 std::vector<float> output_features; in GetCachedBoundsSensitiveFeatures() local
62 &output_features); in GetCachedBoundsSensitiveFeatures()
63 return output_features; in GetCachedBoundsSensitiveFeatures()
Dfeature-processor.cc832 std::vector<float>* output_features) const { in AppendTokenFeaturesWithCache()
848 output_features->insert(output_features->end(), it->second.begin(), in AppendTokenFeaturesWithCache()
850 output_features->insert(output_features->end(), dense_features.begin(), in AppendTokenFeaturesWithCache()
868 output_features->resize(output_features->size() + embedding_size); in AppendTokenFeaturesWithCache()
870 output_features->data() + output_features->size(); in AppendTokenFeaturesWithCache()
888 output_features->insert(output_features->end(), dense_features.begin(), in AppendTokenFeaturesWithCache()
Dfeature-processor.h291 std::vector<float>* output_features) const;
/external/libtextclassifier/native/actions/
Dfeature-processor.cc84 std::vector<float>* output_features) const { in AppendFeatures()
87 output_features->resize(output_features->size() + embedding_size); in AppendFeatures()
89 output_features->data() + output_features->size(); in AppendFeatures()
100 output_features->insert(output_features->end(), dense_features.begin(), in AppendFeatures()
107 std::vector<float>* output_features) const { in AppendTokenFeatures()
117 output_features); in AppendTokenFeatures()
123 std::vector<float>* output_features) const { in AppendTokenFeatures()
125 if (!AppendTokenFeatures(token, embedding_executor, output_features)) { in AppendTokenFeatures()
Dfeature-processor.h46 std::vector<float>* output_features) const;
51 std::vector<float>* output_features) const;
57 std::vector<float>* output_features) const;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc228 auto output_features = in TryResolvePaddedShapesForTensorCore() local
230 if (input_features == 3 && (output_features == 32 || output_features == 64)) { in TryResolvePaddedShapesForTensorCore()
Dgpu_conv_rewriter.cc563 int64 output_features = rhs->shape().dimensions(output_feature_dimension); in MatchBackwardInput() local
603 output_features * num_groups); in MatchBackwardInput()