Home
last modified time | relevance | path

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

/external/libtextclassifier/
Dcached-features.cc87 int click_pos, std::vector<float>* output_features) const { in AppendClickContextFeaturesForClick()
94 /*read_mask_span=*/{0, TokenSpanSize(extraction_span_)}, output_features); in AppendClickContextFeaturesForClick()
98 TokenSpan selected_span, std::vector<float>* output_features) const { in AppendBoundsSensitiveFeaturesForSpan()
112 /*read_mask_span=*/{0, selected_span.second}, output_features); in AppendBoundsSensitiveFeaturesForSpan()
122 output_features); in AppendBoundsSensitiveFeaturesForSpan()
125 AppendBagFeatures(selected_span, output_features); in AppendBoundsSensitiveFeaturesForSpan()
129 output_features->push_back( in AppendBoundsSensitiveFeaturesForSpan()
136 std::vector<float>* output_features) const { in AppendFeaturesInternal()
140 AppendPaddingFeatures(output_features); in AppendFeaturesInternal()
142 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.cc937 std::vector<float>* output_features) const { in AppendTokenFeaturesWithCache()
953 output_features->insert(output_features->end(), it->second.begin(), in AppendTokenFeaturesWithCache()
955 output_features->insert(output_features->end(), dense_features.begin(), in AppendTokenFeaturesWithCache()
973 output_features->resize(output_features->size() + embedding_size); in AppendTokenFeaturesWithCache()
975 output_features->data() + output_features->size(); in AppendTokenFeaturesWithCache()
993 output_features->insert(output_features->end(), dense_features.begin(), in AppendTokenFeaturesWithCache()
Dfeature-processor.h303 std::vector<float>* output_features) const;
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc384 const int64 output_features = in HandleConvolution() local
390 output_features > 0 in HandleConvolution()
391 ? ShapeUtil::ElementsIn(rhs_instruction->shape()) / output_features in HandleConvolution()