Home
last modified time | relevance | path

Searched refs:sparse_features (Results 1 – 15 of 15) sorted by relevance

/external/libtextclassifier/native/utils/
Dtoken-feature-extractor_test.cc46 std::vector<int> sparse_features; in TEST_F() local
49 extractor.Extract(Token{"Hello", 0, 5}, true, &sparse_features, in TEST_F()
52 EXPECT_THAT(sparse_features, in TEST_F()
75 sparse_features.clear(); in TEST_F()
77 extractor.Extract(Token{"world!", 23, 29}, false, &sparse_features, in TEST_F()
80 EXPECT_THAT(sparse_features, in TEST_F()
116 std::vector<int> sparse_features; in TEST_F() local
119 extractor.Extract(Token{"Hello", 0, 5}, true, &sparse_features, in TEST_F()
122 EXPECT_THAT(sparse_features, in TEST_F()
126 sparse_features.clear(); in TEST_F()
[all …]
Dtoken-feature-extractor.cc83 std::vector<int>* sparse_features, in Extract() argument
88 if (sparse_features) { in Extract()
89 *sparse_features = ExtractCharactergramFeatures(token); in Extract()
Dtoken-feature-extractor.h79 std::vector<int>* sparse_features,
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.cc66 const Example::SparseFeatures& sparse_features, in UpdateSparseDeltaWeights() argument
68 for (int64 k = 0; k < sparse_features.indices->size(); ++k) { in UpdateSparseDeltaWeights()
70 sparse_features.values == nullptr ? 1.0 : (*sparse_features.values)(k); in UpdateSparseDeltaWeights()
71 auto it = indices_to_id_.find((*sparse_features.indices)(k)); in UpdateSparseDeltaWeights()
165 const Example::SparseFeatures& sparse_features = sparse_features_[j]; in ComputeWxAndWeightedExampleNorm() local
169 for (int64 k = 0; k < sparse_features.indices->size(); ++k) { in ComputeWxAndWeightedExampleNorm()
170 const int64 feature_index = (*sparse_features.indices)(k); in ComputeWxAndWeightedExampleNorm()
171 const double feature_value = sparse_features.values == nullptr in ComputeWxAndWeightedExampleNorm()
173 : (*sparse_features.values)(k); in ComputeWxAndWeightedExampleNorm()
414 Example::SparseFeatures* const sparse_features = in CreateSparseFeatureRepresentation() local
[all …]
Dsdca_internal.h264 const Example::SparseFeatures& sparse_features,
/external/libtextclassifier/native/actions/
Dfeature-processor_test.cc35 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding() argument
38 EXPECT_THAT(sparse_features, SizeIs(1)); in AddEmbedding()
39 dest[0] = sparse_features.data()[0]; in AddEmbedding()
40 dest[1] = sparse_features.data()[0]; in AddEmbedding()
41 dest[2] = -sparse_features.data()[0]; in AddEmbedding()
42 dest[3] = -sparse_features.data()[0]; in AddEmbedding()
Dfeature-processor.cc81 const std::vector<int>& sparse_features, in AppendFeatures() argument
91 TensorView<int>(sparse_features.data(), in AppendFeatures()
92 {static_cast<int>(sparse_features.size())}), in AppendFeatures()
109 std::vector<int> sparse_features; in AppendTokenFeatures() local
112 &sparse_features, &dense_features)) { in AppendTokenFeatures()
116 return AppendFeatures(sparse_features, dense_features, embedding_executor, in AppendTokenFeatures()
Dfeature-processor.h43 bool AppendFeatures(const std::vector<int>& sparse_features,
Dactions-suggestions_test.cc1386 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding() argument
1389 EXPECT_EQ(sparse_features.size(), 1); in AddEmbedding()
1390 dest[0] = sparse_features.data()[0]; in AddEmbedding()
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_correctness_test.py219 sparse_features = (
234 for sparse in sparse_features:
241 sparse_features = (sparse_features, tuple(weights))
242 return sparse_features
246 sparse_features = self._create_sparse_data(include_weights, weight)
248 dataset = dataset_ops.DatasetV2.from_tensors(sparse_features)
256 sparse_features = self._create_sparse_data(include_weights, weight)
258 sparse_features)
Dtpu_embedding_v2_test.py643 sparse_features = next(sparse_iter)
645 features = (sparse_features[0], ragged_features[1], sparse_features[2])
718 sparse_features = next(sparse_iter)
720 mid_level_api.enqueue(sparse_features, training=False)
748 sparse_features = next(sparse_iter)
749 mid_level_api.enqueue(sparse_features, training=False)
757 sparse_features)
1044 sparse_features = (
1059 for sparse in sparse_features:
1066 sparse_features = (sparse_features, tuple(weights))
[all …]
/external/libtextclassifier/native/annotator/
Dmodel-executor.cc153 const TensorView<int>& sparse_features, float* dest, int dest_size) const { in AddEmbedding() argument
159 const int num_sparse_features = sparse_features.size(); in AddEmbedding()
161 const int bucket_id = sparse_features.data()[i]; in AddEmbedding()
Dmodel-executor.h70 virtual bool AddEmbedding(const TensorView<int>& sparse_features, float* dest,
86 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest,
Dfeature-processor_test.cc65 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding() argument
68 EXPECT_EQ(sparse_features.size(), 1); in AddEmbedding()
69 dest[0] = sparse_features.data()[0]; in AddEmbedding()
70 dest[1] = sparse_features.data()[0]; in AddEmbedding()
71 dest[2] = -sparse_features.data()[0]; in AddEmbedding()
72 dest[3] = -sparse_features.data()[0]; in AddEmbedding()
Dfeature-processor.cc857 std::vector<int> sparse_features; in AppendTokenFeaturesWithCache() local
861 &sparse_features, &dense_features)) { in AppendTokenFeaturesWithCache()
872 TensorView<int>(sparse_features.data(), in AppendTokenFeaturesWithCache()
873 {static_cast<int>(sparse_features.size())}), in AppendTokenFeaturesWithCache()