Home
last modified time | relevance | path

Searched refs:sparse_keys (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dparsing_ops.py205 sparse_keys = []
224 sparse_keys.append(key)
246 if index_key in sparse_keys:
247 dtype = sparse_types[sparse_keys.index(index_key)]
252 sparse_keys.append(index_key)
254 if feature.value_key in sparse_keys:
255 dtype = sparse_types[sparse_keys.index(feature.value_key)]
260 sparse_keys.append(feature.value_key)
301 sparse_keys, sparse_types, dense_keys, dense_types, dense_defaults,
800 (sparse_keys, sparse_types, dense_keys, dense_types, dense_defaults,
[all …]
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dparsing_ops.py42 (sparse_keys, sparse_types, dense_keys, dense_types, dense_defaults,
49 (_, dense_defaults_vec, sparse_keys, sparse_types, dense_keys, dense_shapes,
51 None, dense_defaults, sparse_keys, sparse_types, dense_keys,
54 self._sparse_keys = sparse_keys
65 for _ in range(len(sparse_keys))]
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops_test.cc151 std::vector<NodeBuilder::NodeOut> sparse_keys; in ParseExample() local
174 sparse_keys.emplace_back(test::graph::Constant(g, key)); in ParseExample()
184 .Input(sparse_keys) in ParseExample()
203 std::vector<string> sparse_keys; in ParseSingleExample() local
225 sparse_keys.push_back(key), sparse_types.push_back(opt.filler.dtype); in ParseSingleExample()
234 .Attr<int64>("num_sparse", sparse_keys.size()) in ParseSingleExample()
235 .Attr("sparse_keys", sparse_keys) in ParseSingleExample()
Dexample_parsing_ops.cc47 OpInputList sparse_keys; in Compute() local
54 OP_REQUIRES_OK(ctx, ctx->input_list("sparse_keys", &sparse_keys)); in Compute()
62 CHECK_EQ(sparse_keys.size(), attrs_.num_sparse); in Compute()
69 sparse_keys_t[di] = sparse_keys[di].scalar<string>()(); in Compute()
233 for (int d = 0; d < attrs_.sparse_keys.size(); ++d) { in Compute()
234 config.sparse.push_back({attrs_.sparse_keys[d], attrs_.sparse_types[d]}); in Compute()
253 for (int d = 0; d < attrs_.sparse_keys.size(); ++d) { in Compute()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dparse_single_example_vectorizer.cc63 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/fuzzing/
Dexample_proto_fast_parsing_fuzz.cc39 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/
Dapi_def_ParseSingleExample.pbtxt28 must match the lengths of `sparse_keys` and `sparse_types`.
32 name: "sparse_keys"
49 Feature given in sparse_keys.
Dapi_def_ExperimentalParseExampleDataset.pbtxt12 name: "sparse_keys"
28 A list of `DTypes` of the same length as `sparse_keys`.
Dapi_def_ParseExample.pbtxt21 name: "sparse_keys"
53 given in sparse_keys.
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h215 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_keys", &sparse_keys)); in Init()
223 if (num_sparse != sparse_keys.size() || num_sparse != sparse_types.size()) { in Init()
226 sparse_keys.size(), ") and sparse_types (", sparse_types.size(), ")"); in Init()
263 std::vector<string> sparse_keys; variable
Dexample_proto_helper.cc428 if (sparse_keys.size() != sparse_types.size()) { in FinishInit()
/external/tensorflow/tensorflow/core/ops/
Dparsing_ops.cc118 for (int i = 0; i < attrs.sparse_keys.size(); ++i) { in __anonb04296470302()
121 for (int i = 0; i < attrs.sparse_keys.size(); ++i) { in __anonb04296470302()
124 for (int i = 0; i < attrs.sparse_keys.size(); ++i) { in __anonb04296470302()
Dops.pbtxt11812 name: "sparse_keys"
22224 name: "sparse_keys"
22524 name: "sparse_keys"
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc157 std::vector<Tensor> dense_defaults, std::vector<string> sparse_keys, in Dataset() argument
169 sparse_keys_(std::move(sparse_keys)), in Dataset()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py2773 sparse_keys = pfor_input.get_attr("sparse_keys")
2781 sparse_keys=sparse_keys,
/external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/
Dsaved_model.pbtxt201 name: "sparse_keys"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt1137 …argspec: "args=[\'input_dataset\', \'num_parallel_calls\', \'dense_defaults\', \'sparse_keys\', \'…
2297 …argspec: "args=[\'serialized\', \'names\', \'sparse_keys\', \'dense_keys\', \'dense_defaults\', \'…
2305 …argspec: "args=[\'serialized\', \'dense_defaults\', \'num_sparse\', \'sparse_keys\', \'dense_keys\…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt1137 …argspec: "args=[\'input_dataset\', \'num_parallel_calls\', \'dense_defaults\', \'sparse_keys\', \'…
2297 …argspec: "args=[\'serialized\', \'names\', \'sparse_keys\', \'dense_keys\', \'dense_defaults\', \'…
2305 …argspec: "args=[\'serialized\', \'dense_defaults\', \'num_sparse\', \'sparse_keys\', \'dense_keys\…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go11029 func ParseExample(scope *Scope, serialized tf.Output, names tf.Output, sparse_keys []tf.Output, den…
11037 …serialized, names, tf.OutputList(sparse_keys), tf.OutputList(dense_keys), tf.OutputList(dense_defa…
16343 …t tf.Output, num_parallel_calls tf.Output, dense_defaults []tf.Output, sparse_keys []string, dense…
16347 …attrs := map[string]interface{}{"sparse_keys": sparse_keys, "dense_keys": dense_keys, "sparse_type…
16838 …e, serialized tf.Output, dense_defaults []tf.Output, num_sparse int64, sparse_keys []string, dense…
16842 …attrs := map[string]interface{}{"num_sparse": num_sparse, "sparse_keys": sparse_keys, "dense_keys"…
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc775 key: "sparse_keys" in CreateImagenetDatasetFunctions()
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt24500 name: "sparse_keys"
24570 name: "sparse_keys"
43955 name: "sparse_keys"
44255 name: "sparse_keys"
Dops_history.v2.pbtxt24129 name: "sparse_keys"
24199 name: "sparse_keys"
43432 name: "sparse_keys"
43732 name: "sparse_keys"
Dops_history.v0.pbtxt19574 name: "sparse_keys"