| /external/libtextclassifier/native/utils/ |
| D | token-feature-extractor_test.cc | 47 std::vector<float> dense_features; in TEST_F() local
50 &dense_features); in TEST_F()
73 EXPECT_THAT(dense_features, testing::ElementsAreArray({1.0, 1.0})); in TEST_F()
76 dense_features.clear(); in TEST_F()
78 &dense_features); in TEST_F()
104 EXPECT_THAT(dense_features, testing::ElementsAreArray({-1.0, 0.0})); in TEST_F()
117 std::vector<float> dense_features; in TEST_F() local
120 &dense_features); in TEST_F()
124 EXPECT_THAT(dense_features, testing::ElementsAreArray({1.0, 1.0})); in TEST_F()
127 dense_features.clear(); in TEST_F()
[all …]
|
| D | token-feature-extractor.cc | 84 std::vector<float>* dense_features) const { in Extract()
85 if (!dense_features) { in Extract()
91 *dense_features = ExtractDenseFeatures(token, is_in_span); in Extract()
106 std::vector<float> dense_features; in ExtractDenseFeatures() local
113 dense_features.push_back(1.0); in ExtractDenseFeatures()
115 dense_features.push_back(-1.0); in ExtractDenseFeatures()
119 dense_features.push_back(1.0); in ExtractDenseFeatures()
121 dense_features.push_back(-1.0); in ExtractDenseFeatures()
128 dense_features.push_back(1.0); in ExtractDenseFeatures()
131 dense_features.push_back(-1.0); in ExtractDenseFeatures()
[all …]
|
| D | token-feature-extractor.h | 80 std::vector<float>* dense_features) const;
|
| /external/libtextclassifier/native/actions/ |
| D | feature-processor.cc | 82 const std::vector<float>& dense_features, in AppendFeatures() argument
100 output_features->insert(output_features->end(), dense_features.begin(), in AppendFeatures()
101 dense_features.end()); in AppendFeatures()
110 std::vector<float> dense_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 | 44 const std::vector<float>& dense_features,
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | dense_features_test.py | 36 from tensorflow.python.keras.feature_column import dense_features as df
57 dense_features = df.DenseFeatures(fc.numeric_column('a'))
58 inputs = self.evaluate(dense_features(features))
88 dense_features = df.DenseFeatures([embedding_column])
91 inputs = dense_features(features)
92 variables = dense_features.variables
102 _ = dense_features(features)
104 self.assertIs(variables[0], dense_features.variables[0])
137 dense_features = df.DenseFeatures(
142 inputs = dense_features(features)
[all …]
|
| D | dense_features_v2_test.py | 52 dense_features = df.DenseFeatures(fc.numeric_column('a'))
53 inputs = self.evaluate(dense_features(features))
83 dense_features = df.DenseFeatures([embedding_column])
86 inputs = dense_features(features)
87 variables = dense_features.variables
97 _ = dense_features(features)
99 self.assertIs(variables[0], dense_features.variables[0])
128 dense_features = df.DenseFeatures([embedding_column])
132 matrix = dense_features(features)
267 dense_features = df.DenseFeatures([price1, price2])
[all …]
|
| D | BUILD | 23 ":dense_features",
43 name = "dense_features",
45 "dense_features.py",
66 ":dense_features",
78 ":dense_features",
168 ":dense_features",
|
| D | dense_features_v2.py | 24 from tensorflow.python.keras.feature_column import dense_features
29 class DenseFeatures(dense_features.DenseFeatures):
|
| D | sequence_feature_column_integration_test.py | 31 from tensorflow.python.keras.feature_column import dense_features
101 dense_input_layer = dense_features.DenseFeatures(ctx_cols)
142 non_seq_input = dense_features.DenseFeatures([shared_non_seq])(features)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column_v2_test.py | 194 dense_features = fc_lib.DenseFeatures([embedding_column])
196 embedding_lookup = dense_features(features)
418 dense_features = fc_lib.DenseFeatures([embedding_column_a])
420 embedding_lookup_a = dense_features(input_features)
499 dense_features = fc_lib.DenseFeatures(embedding_column)
503 dense_features(input_features)
520 dense_features = fc_lib.DenseFeatures(embedding_column)
525 dense_features(input_features)
608 dense_features = fc_lib.DenseFeatures(embedding_column)
613 embedding_lookup = dense_features(input_features)
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | prediction_ops.cc | 41 auto dense_features = dense_features_t->matrix<float>(); in Compute() local
62 auto traverse = [this, &out, &dense_features, &decision_tree_resource, in Compute()
68 decision_tree_resource->TraverseTree(example_id, &dense_features); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops_test.cc | 192 std::vector<NodeBuilder::NodeOut> dense_features; in GetGraphs() local
193 dense_features.reserve(num_dense_feature_groups); in GetGraphs()
195 dense_features.push_back(NodeBuilder::NodeOut( in GetGraphs()
217 .Input(dense_features) in GetGraphs()
|
| D | sdca_internal.cc | 477 auto dense_features = dense_features_inputs[i].template matrix<float>(); in CreateDenseFeatureRepresentation() local
480 new Example::DenseVector{dense_features, example_id}); in CreateDenseFeatureRepresentation()
482 if (!weights.DenseIndexValid(i, dense_features.dimension(1) - 1)) { in CreateDenseFeatureRepresentation()
486 dense_features.dimension(1)); in CreateDenseFeatureRepresentation()
|
| /external/libtextclassifier/native/annotator/ |
| D | feature-processor.cc | 839 std::vector<float> dense_features; in AppendTokenFeaturesWithCache() local
842 /*sparse_features=*/nullptr, &dense_features)) { in AppendTokenFeaturesWithCache()
850 output_features->insert(output_features->end(), dense_features.begin(), in AppendTokenFeaturesWithCache()
851 dense_features.end()); in AppendTokenFeaturesWithCache()
858 std::vector<float> dense_features; in AppendTokenFeaturesWithCache() local
861 &sparse_features, &dense_features)) { in AppendTokenFeaturesWithCache()
888 output_features->insert(output_features->end(), dense_features.begin(), in AppendTokenFeaturesWithCache()
889 dense_features.end()); in AppendTokenFeaturesWithCache()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TensorForestTreePredict.pbtxt | 8 name: "dense_features"
|
| D | SdcaOptimizer.pbtxt | 19 name: "dense_features"
137 name: "dense_features"
|
| D | SdcaOptimizerV2.pbtxt | 19 name: "dense_features"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TensorForestTreePredict.pbtxt | 8 name: "dense_features"
|
| D | SdcaOptimizer.pbtxt | 19 name: "dense_features"
137 name: "dense_features"
|
| D | SdcaOptimizerV2.pbtxt | 19 name: "dense_features"
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_lib.py | 29 from tensorflow.python.keras.feature_column.dense_features import *
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TensorForestTreePredict.pbtxt | 17 name: "dense_features"
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils_test.py | 43 from tensorflow.python.keras.feature_column import dense_features
161 model = sequential.Sequential([dense_features.DenseFeatures(columns)])
171 model = sequential.Sequential([dense_features.DenseFeatures(columns)])
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | serialization.py | 139 …from tensorflow.python.keras.feature_column.dense_features import DenseFeatures # pylint: disable…
|