HomeSort by relevance Sort by last modified time
    Searched refs:VectorSpan (Results 1 - 10 of 10) sorted by null

  /external/libtextclassifier/common/
vector-span.h 26 class VectorSpan {
28 VectorSpan() : begin_(), end_() {}
29 VectorSpan(const std::vector<T>& v) // NOLINT(runtime/explicit)
31 VectorSpan(typename std::vector<T>::const_iterator begin,
embedding-network.h 196 bool ComputeLogits(const VectorSpan<float> &input, Vector *scores) const;
211 bool ComputeLogitsInternal(const VectorSpan<float> &concat,
217 bool FinishComputeFinalScoresInternal(const VectorSpan<float> &concat,
embedding-network.cc 89 const VectorSpan<float> &x,
234 bool EmbeddingNetwork::ComputeLogits(const VectorSpan<float> &input,
244 bool EmbeddingNetwork::ComputeLogitsInternal(const VectorSpan<float> &input,
251 const VectorSpan<float> &input, Vector *scores) const {
259 VectorSpan<float> h_in(input);
269 h_in = VectorSpan<float>(*h_out);
  /external/libtextclassifier/smartselect/
cached-features.cc 34 bool CachedFeatures::Get(int click_pos, VectorSpan<float>* features,
35 VectorSpan<Token>* output_tokens) {
44 VectorSpan<float>(features_.begin() + token_start * feature_vector_size_,
46 *output_tokens = VectorSpan<Token>(tokens_.begin() + token_start,
55 void CachedFeatures::RemapV0FeatureVector(VectorSpan<float>* features) {
79 *features = VectorSpan<float>(v0_feature_storage_);
cached-features.h 42 CachedFeatures(VectorSpan<Token> tokens, int context_size,
57 // Gets a VectorSpan with the features for given click position.
58 bool Get(int click_pos, VectorSpan<float>* features,
59 VectorSpan<Token>* output_tokens);
64 // VectorSpan<float>, so the output of Extract is valid only until the next
91 void RemapV0FeatureVector(VectorSpan<float>* features);
94 const VectorSpan<Token> tokens_;
feature-processor.h 151 VectorSpan<Token> tokens,
185 bool LabelToSpan(int label, const VectorSpan<Token>& output_tokens,
text-classification-model.cc 252 VectorSpan<float> features;
253 VectorSpan<Token> output_tokens;
392 VectorSpan<float> features;
393 VectorSpan<Token> output_tokens;
feature-processor.cc 209 const int label, const VectorSpan<Token>& tokens,
424 const VectorSpan<Token> tokens,
  /external/libtextclassifier/tests/
cached-features_test.cc 63 VectorSpan<float> features;
64 VectorSpan<Token> output_tokens;
95 VectorSpan<float> features;
96 VectorSpan<Token> output_tokens;
130 VectorSpan<float> features;
133 features = VectorSpan<float>(features_orig);
141 features = VectorSpan<float>(features_orig);
embedding-network_test.cc 80 VectorSpan<float>(input), &output);

Completed in 138 milliseconds