| /external/libtextclassifier/utils/ |
| D | token-feature-extractor_test.cc | 46 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 …]
|
| D | token-feature-extractor.cc | 83 std::vector<int>* sparse_features, in Extract() argument
88 if (sparse_features) { in Extract()
89 *sparse_features = ExtractCharactergramFeatures(token); in Extract()
|
| D | token-feature-extractor.h | 77 std::vector<int>* sparse_features,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_internal.cc | 66 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 …]
|
| D | sdca_internal.h | 264 const Example::SparseFeatures& sparse_features,
|
| /external/libtextclassifier/actions/ |
| D | feature-processor_test.cc | 35 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding() argument
38 EXPECT_EQ(sparse_features.size(), 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()
|
| D | feature-processor.cc | 81 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()
|
| D | feature-processor.h | 43 bool AppendFeatures(const std::vector<int>& sparse_features,
|
| D | actions-suggestions_test.cc | 1037 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest, in AddEmbedding() argument
1040 EXPECT_EQ(sparse_features.size(), 1); in AddEmbedding()
1041 dest[0] = sparse_features.data()[0]; in AddEmbedding()
|
| /external/tensorflow/tensorflow/contrib/libsvm/python/kernel_tests/ |
| D | decode_libsvm_op_test.py | 37 sparse_features, labels = libsvm_ops.decode_libsvm(
40 sparse_features, validate_indices=False)
55 sparse_features, labels = libsvm_ops.decode_libsvm(
58 sparse_features, validate_indices=False)
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ |
| D | tensor_forest.py | 475 sparse_features=processed_sparse_features,
514 sparse_features=processed_sparse_features,
587 sparse_features=None, argument
616 if sparse_features is not None:
617 sparse_indices = sparse_features.indices
618 sparse_values = sparse_features.values
619 sparse_shape = sparse_features.dense_shape
661 def inference_graph(self, input_data, data_spec, sparse_features=None): argument
676 if sparse_features is not None:
677 sparse_indices = sparse_features.indices
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
| D | data_ops.py | 125 sparse_features = []
138 sparse_features.append(
161 if sparse_features:
162 processed_sparse_features = sparse_ops.sparse_concat(1, sparse_features)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 583 sparse_features, device_ordinal=i % self._num_cores_per_host)
584 for i, sparse_features in enumerate(sparse_features_list)
599 for i, sparse_features in enumerate(sparse_features_list):
600 used_feature_set = set(sparse_features.keys())
617 for feature, tensor in six.iteritems(sparse_features):
649 def _generate_enqueue_op(self, sparse_features, device_ordinal): argument
650 with ops.colocate_with(list(sparse_features.values())[0]):
652 self._format_for_tpu_embedding_sparse_tensor_batch(sparse_features))
661 def _format_for_tpu_embedding_sparse_tensor_batch(self, sparse_features): argument
677 tensor = sparse_features[feature]
|
| D | _tpu_estimator_embedding.py | 278 sparse_features = collections.OrderedDict()
282 sparse_features[feature_key] = features.pop(feature_key)
284 return features, labels, sparse_features
|
| D | tpu_estimator.py | 898 features, labels, sparse_features = (
900 sparse_features_list.append(sparse_features)
|
| /external/libtextclassifier/annotator/ |
| D | model-executor.cc | 153 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()
|
| D | model-executor.h | 70 virtual bool AddEmbedding(const TensorView<int>& sparse_features, float* dest,
86 bool AddEmbedding(const TensorView<int>& sparse_features, float* dest,
|
| D | feature-processor_test.cc | 65 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()
|
| D | feature-processor.cc | 827 std::vector<int> sparse_features; in AppendTokenFeaturesWithCache() local
831 &sparse_features, &dense_features)) { in AppendTokenFeaturesWithCache()
842 TensorView<int>(sparse_features.data(), in AppendTokenFeaturesWithCache()
843 {static_cast<int>(sparse_features.size())}), in AppendTokenFeaturesWithCache()
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_optimizer.py | 149 dense_features, sparse_features, sparse_feature_with_values = [], [], []
252 sparse_feature_with_values.extend(sparse_features)
255 sparse_features=sparse_feature_with_values,
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/kernel_tests/ |
| D | sdca_ops_test.py | 87 sparse_features = [
106 sparse_features=sparse_features,
116 sparse_features = [
125 sparse_features=sparse_features,
197 sparse_features=[],
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dnn_linear_combined_test.py | 484 sparse_features = [
492 sparse_features[0], dimension=1)
509 linear_feature_columns=sparse_features,
|
| D | linear_test.py | 481 sparse_features = [
502 feature_columns=sparse_features, config=config)
|