| /external/tensorflow/tensorflow/python/util/ |
| D | example_parser_configuration.py | 50 num_sparse = parse_example_op.get_attr("Nsparse")
52 total_features = num_dense + num_sparse
58 if len(sparse_types) != num_sparse:
61 (len(sparse_types), num_sparse))
90 dense_keys_start = sparse_keys_start + num_sparse
95 sparse_values_start = num_sparse
96 sparse_shapes_start = sparse_values_start + num_sparse
97 dense_values_start = sparse_shapes_start + num_sparse
120 for i in range(num_sparse):
140 num_sparse = parse_example_op.get_attr("num_sparse")
[all …]
|
| /external/tensorflow/tensorflow/core/example/ |
| D | example_parser_configuration.cc | 58 auto num_sparse = attr_map.at("Nsparse").i(); in ExtractExampleParserConfiguration() local
61 var_len_features->resize(num_sparse); in ExtractExampleParserConfiguration()
80 if (sparse_types.list().type_size() != num_sparse) { in ExtractExampleParserConfiguration()
83 " elements != NSparse attr: ", num_sparse); in ExtractExampleParserConfiguration()
112 int dense_keys_start = sparse_keys_start + num_sparse; in ExtractExampleParserConfiguration()
115 for (int i = 0; i < num_sparse; ++i) { in ExtractExampleParserConfiguration()
133 int sparse_values_output_start = sparse_indices_output_start + num_sparse; in ExtractExampleParserConfiguration()
134 int sparse_shapes_output_start = sparse_values_output_start + num_sparse; in ExtractExampleParserConfiguration()
135 int dense_values_output_start = sparse_shapes_output_start + num_sparse; in ExtractExampleParserConfiguration()
139 for (int i = 0; i < num_sparse; ++i) { in ExtractExampleParserConfiguration()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.h | 173 TF_RETURN_IF_ERROR(ctx->GetAttr("Nsparse", &num_sparse));
179 TF_RETURN_IF_ERROR(ctx->GetAttr("num_sparse", &num_sparse));
189 int64 num_sparse; member
216 int num_sparse; in Init() local
217 TF_RETURN_IF_ERROR(ctx->GetAttr("num_sparse", &num_sparse)); in Init()
218 if (num_sparse != sparse_keys.size() || num_sparse != sparse_types.size()) { in Init()
220 "num_sparse (", num_sparse, ") must match the size of sparse_keys (", in Init()
|
| D | example_proto_helper.cc | 420 if (static_cast<size_t>(num_sparse) != sparse_types.size()) { in FinishInit()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ParseSingleExample.pbtxt | 14 number_attr: "num_sparse"
23 number_attr: "num_sparse"
30 name: "num_sparse"
|
| D | ParseExampleV2.pbtxt | 30 number_attr: "num_sparse"
39 number_attr: "num_sparse"
66 name: "num_sparse"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ParseSingleExample.pbtxt | 14 number_attr: "num_sparse"
23 number_attr: "num_sparse"
30 name: "num_sparse"
|
| D | ParseExampleV2.pbtxt | 30 number_attr: "num_sparse"
39 number_attr: "num_sparse"
66 name: "num_sparse"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops_test.cc | 88 auto set_outputs = [&op](int num_sparse, int num_dense, in TEST()
99 .Input(NodeOutList(num_sparse, string_in)) in TEST()
102 .Attr("sparse_types", DataTypeList(num_sparse, DT_FLOAT)) in TEST()
305 auto set_outputs = [&op](int num_sparse, int num_dense, int num_ragged, in TEST()
320 .Attr("num_sparse", num_sparse) in TEST()
321 .Attr("sparse_types", DataTypeList(num_sparse, DT_FLOAT)) in TEST()
|
| D | ragged_array_ops.cc | 81 int num_sparse; in __anoncd0b9d3b0102() local
82 TF_RETURN_IF_ERROR(c->GetAttr("Nsparse", &num_sparse)); in __anoncd0b9d3b0102()
99 int dense_start = num_ragged * 2 + num_sparse * 3; in __anoncd0b9d3b0102()
|
| D | parsing_ops.cc | 45 void AddSparseOutputShapes(int num_sparse, const ShapeHandle input_shape, in AddSparseOutputShapes() argument
53 for (int i = 0; i < num_sparse; ++i) { // sparse_indices in AddSparseOutputShapes()
56 for (int i = 0; i < num_sparse; ++i) { // sparse_values in AddSparseOutputShapes()
59 for (int i = 0; i < num_sparse; ++i) { // sparse_dense_shapes in AddSparseOutputShapes()
170 AddSparseOutputShapes(attrs.num_sparse, input, 1, c, &output_idx); in __anonb04296470402()
218 AddSparseOutputShapes(attrs.num_sparse, input, 1, c, &output_idx); in __anonb04296470502()
|
| D | ops.pbtxt | 28585 number_attr: "num_sparse"
28594 number_attr: "num_sparse"
28621 name: "num_sparse"
29146 number_attr: "num_sparse"
29155 number_attr: "num_sparse"
29162 name: "num_sparse"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParseSingleExample.pbtxt | 25 name: "num_sparse"
34 A list of `num_sparse` strings.
48 A list of `num_sparse` types; the data types of data in each
|
| D | api_def_ParseExampleV2.pbtxt | 58 name: "num_sparse"
66 A list of `num_sparse` types; the data types of data in each Feature
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_cross_op.cc | 298 int num_sparse; in RaggedCrossOp() local
299 OP_REQUIRES_OK(context, context->GetAttr("Nsparse", &num_sparse)); in RaggedCrossOp()
313 OP_REQUIRES(context, num_sparse == sparse_values_types_.size(), in RaggedCrossOp()
383 const auto num_sparse = sparse_indices_list.size(); in ValidateInput() local
396 for (int i = 0; i < num_sparse; ++i) { in ValidateInput()
423 for (int i = 0; i < num_sparse; ++i) { in ValidateInput()
|
| D | example_parsing_ops.cc | 156 if (sparse_keys_t.size() != attrs_.num_sparse) { in CheckInputShapes()
159 sparse_keys_t.size(), " vs. ", attrs_.num_sparse); in CheckInputShapes()
221 config.sparse.reserve(attrs_.num_sparse); in MakeConfig()
222 for (int d = 0; d < attrs_.num_sparse; ++d) { in MakeConfig()
268 for (int d = 0; d < attrs_.num_sparse; ++d) { in WriteOutput()
|
| D | example_parsing_ops_test.cc | 260 int num_sparse = opt.benchmark_type == kSparse ? num_keys : 0; in ParseExampleV2() local
270 .Attr("num_sparse", num_sparse) in ParseExampleV2()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | parse_example_v2.mlir | 14 …pe$DT_FLOAT"], dense_shapes = [#tf.shape<>, #tf.shape<>], device = "", num_sparse = 2 : i64, ragge…
45 // CHECK-NEXT: key: "num_sparse"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | parse_example_v2.pbtxt | 194 key: "num_sparse"
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 730 int num_sparse = 0; in PrepareParseExample() local
736 num_sparse = value.AsInt32(); in PrepareParseExample()
742 data->sparse_size = num_sparse; in PrepareParseExample()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_ops.py | 357 num_sparse=len(params.sparse_keys),
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 3524 …fault_0, %dense_default_1) {dense_shapes = [#tf.shape<>, #tf.shape<>], num_sparse = 0 : i64, resul…
3533 …fault_0, %dense_default_1) {dense_shapes = [#tf.shape<>, #tf.shape<>], num_sparse = 0 : i64, resul…
3542 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 2 : i64, resul…
3550 …// expected-error @+1 {{attribute 'num_sparse' should be the same as the length of attribute 'spar…
3551 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 3 : i64, resul…
3560 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 2 : i64, resul…
3569 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 2 : i64, resul…
3578 …empty_str_vector, %empty_str_vector, %ragged_keys) {dense_shapes = [], num_sparse = 0 : i64, resul…
3587 …empty_str_vector, %empty_str_vector, %ragged_keys) {dense_shapes = [], num_sparse = 0 : i64, resul…
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 4123 num_sparse = pfor_input.get_attr("num_sparse")
4138 num_sparse=num_sparse,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 372 if (op.num_sparse() != sparse_types_count) { in Verify()
|
| D | tf_ops.td | 516 Confined<I64Attr, [IntMinValue<0>]>:$num_sparse,
|