| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example_test.cc | 157 std::vector<std::string> sparse_keys, in ParseExampleOpModel() argument
168 std::for_each(sparse_keys.begin(), sparse_keys.end(), [&](auto&&) { in ParseExampleOpModel()
178 if (!sparse_keys.empty()) { in ParseExampleOpModel()
179 for (int i = 0; i < sparse_keys.size(); i++) { in ParseExampleOpModel()
182 for (int i = 0; i < sparse_keys.size(); i++) { in ParseExampleOpModel()
185 for (int i = 0; i < sparse_keys.size(); i++) { in ParseExampleOpModel()
213 for (const auto& key : sparse_keys) { in ParseExampleOpModel()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 353 sparse_keys=params.sparse_keys,
357 num_sparse=len(params.sparse_keys),
374 zip(params.sparse_keys + params.dense_keys + params.ragged_keys,
629 context_sparse_keys=context.sparse_keys,
632 feature_list_sparse_keys=feature_list.sparse_keys,
639 Ncontext_sparse=len(context.sparse_keys),
640 Nfeature_list_sparse=len(feature_list.sparse_keys),
684 context.sparse_keys + context.dense_keys + context.ragged_keys,
689 feature_list.sparse_keys + feature_list.dense_keys +
|
| D | parsing_config.py | 392 sparse_keys=None, argument
405 sparse_keys = [] if sparse_keys is None else sparse_keys
416 self.sparse_keys = sparse_keys
463 return len(self.dense_keys) + len(self.sparse_keys) + len(self.ragged_keys)
533 if key in self.sparse_keys:
534 original_dtype = self.sparse_types[self.sparse_keys.index(key)]
539 self.sparse_keys.append(key)
632 if len(self.sparse_types) != len(self.sparse_keys):
635 (len(self.sparse_types), len(self.sparse_keys)))
646 sparse_key_set = set(self.sparse_keys)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | parse_single_example_vectorizer.cc | 63 std::vector<NodeBuilder::NodeOut> sparse_keys; in Vectorize() local
64 TF_RETURN_IF_ERROR(make_list_input_from_attr("sparse_keys", &sparse_keys)); in Vectorize()
76 .Input(sparse_keys) in Vectorize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops_test.cc | 160 std::vector<NodeBuilder::NodeOut> sparse_keys; in ParseExample() local
183 sparse_keys.emplace_back(test::graph::Constant(g, key)); in ParseExample()
193 .Input(sparse_keys) in ParseExample()
256 auto& sparse_keys = (bm_type == kSparse) ? keys_t : empty_keys; in ParseExampleV2() local
266 .Input(test::graph::Constant(g, sparse_keys)) in ParseExampleV2()
290 std::vector<string> sparse_keys; in ParseSingleExample() local
312 sparse_keys.push_back(key), sparse_types.push_back(opt.filler.dtype); in ParseSingleExample()
321 .Attr<int64>("num_sparse", sparse_keys.size()) in ParseSingleExample()
322 .Attr("sparse_keys", sparse_keys) in ParseSingleExample()
|
| D | example_parsing_ops.cc | 301 metrics::RecordParseSparseFeature(attrs_.sparse_keys.size()); in ParseSingleExampleOp()
360 for (int d = 0; d < attrs_.sparse_keys.size(); ++d) { in Compute()
361 config.sparse.push_back({attrs_.sparse_keys[d], attrs_.sparse_types[d]}); in Compute()
380 for (int d = 0; d < attrs_.sparse_keys.size(); ++d) { in Compute()
569 const Tensor* dense_keys, const Tensor* sparse_keys, in MakeContextConfig() argument
580 sparse_keys in MakeContextConfig()
581 ? gtl::ArraySlice<tstring>(sparse_keys->flat<tstring>().data(), in MakeContextConfig()
628 const Tensor* dense_keys, const Tensor* sparse_keys, in MakeFeatureListConfig() argument
639 sparse_keys in MakeFeatureListConfig()
640 ? gtl::ArraySlice<tstring>(sparse_keys->flat<tstring>().data(), in MakeFeatureListConfig()
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | example_proto_fast_parsing_fuzz.cc | 39 std::vector<Output> sparse_keys; // Empty. in BuildGraph() local
47 sparse_keys, dense_keys, dense_defaults, sparse_types, in BuildGraph()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParseExampleV2.pbtxt | 22 name: "sparse_keys"
67 given in sparse_keys.
76 given in ragged_keys (where `num_ragged = sparse_keys.size()`).
85 given in ragged_keys (where `num_ragged = sparse_keys.size()`).
|
| D | api_def_ParseSingleExample.pbtxt | 28 must match the lengths of `sparse_keys` and `sparse_types`.
32 name: "sparse_keys"
49 Feature given in sparse_keys.
|
| D | api_def_ExperimentalParseExampleDataset.pbtxt | 12 name: "sparse_keys"
28 A list of `DTypes` of the same length as `sparse_keys`.
|
| D | api_def_ParseExampleDataset.pbtxt | 12 name: "sparse_keys"
28 A list of `DTypes` of the same length as `sparse_keys`.
|
| D | api_def_ParseExampleDatasetV2.pbtxt | 12 name: "sparse_keys"
28 A list of `DTypes` of the same length as `sparse_keys`.
|
| D | api_def_ParseExample.pbtxt | 21 name: "sparse_keys"
53 given in sparse_keys.
|
| /external/tensorflow/tensorflow/python/util/ |
| D | example_parser_configuration.py | 153 sparse_keys = fetched[0].tolist()
157 assert len(sparse_keys) == num_sparse
191 key = sparse_keys[i]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.h | 210 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_keys", &sparse_keys)); in Init()
218 if (num_sparse != sparse_keys.size() || num_sparse != sparse_types.size()) { in Init()
221 sparse_keys.size(), ") and sparse_types (", sparse_types.size(), ")"); in Init()
229 std::vector<tstring> sparse_keys; member
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ExperimentalParseExampleDataset.pbtxt | 20 name: "sparse_keys"
90 name: "sparse_keys"
|
| D | ParseExampleDataset.pbtxt | 20 name: "sparse_keys"
97 name: "sparse_keys"
|
| D | ParseSingleExample.pbtxt | 35 name: "sparse_keys"
|
| D | ParseExample.pbtxt | 12 name: "sparse_keys"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ExperimentalParseExampleDataset.pbtxt | 20 name: "sparse_keys"
90 name: "sparse_keys"
|
| D | ParseSingleExample.pbtxt | 35 name: "sparse_keys"
|
| D | ParseExample.pbtxt | 12 name: "sparse_keys"
|
| D | ParseExampleDatasetV2.pbtxt | 20 name: "sparse_keys"
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | parsing_ops.py | 57 self._sparse_keys = params.sparse_keys
71 for (key, value_type) in zip(params.sparse_keys, params.sparse_types):
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | parse_example_v2.pbtxt | 92 name: "ParseExample/ParseExampleV2/sparse_keys"
168 input: "ParseExample/ParseExampleV2/sparse_keys"
|