| /external/tensorflow/tensorflow/python/util/ |
| D | example_parser_configuration.py | 52 num_sparse = parse_example_op.get_attr("Nsparse")
54 total_features = num_dense + num_sparse
60 if len(sparse_types) != num_sparse:
63 (len(sparse_types), num_sparse))
92 dense_keys_start = sparse_keys_start + num_sparse
97 sparse_values_start = num_sparse
98 sparse_shapes_start = sparse_values_start + num_sparse
99 dense_values_start = sparse_shapes_start + num_sparse
122 for i in range(num_sparse):
142 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 | 174 TF_RETURN_IF_ERROR(ctx->GetAttr("Nsparse", &num_sparse));
180 TF_RETURN_IF_ERROR(ctx->GetAttr("num_sparse", &num_sparse));
190 int64 num_sparse; member
217 int num_sparse; in Init() local
218 TF_RETURN_IF_ERROR(ctx->GetAttr("num_sparse", &num_sparse)); in Init()
219 if (num_sparse != sparse_keys.size() || num_sparse != sparse_types.size()) { in Init()
221 "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 __anone8952ce70102() local
82 TF_RETURN_IF_ERROR(c->GetAttr("Nsparse", &num_sparse)); in __anone8952ce70102()
99 int dense_start = num_ragged * 2 + num_sparse * 3; in __anone8952ce70102()
|
| 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()
172 AddSparseOutputShapes(attrs.num_sparse, input, 1, c, &output_idx); in __anonc1bcaa730402()
220 AddSparseOutputShapes(attrs.num_sparse, input, 1, c, &output_idx); in __anonc1bcaa730502()
|
| /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 | 300 int num_sparse; in RaggedCrossOp() local
301 OP_REQUIRES_OK(context, context->GetAttr("Nsparse", &num_sparse)); in RaggedCrossOp()
315 OP_REQUIRES(context, num_sparse == sparse_values_types_.size(), in RaggedCrossOp()
385 const auto num_sparse = sparse_indices_list.size(); in ValidateInput() local
398 for (int i = 0; i < num_sparse; ++i) { in ValidateInput()
425 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 …T"], dense_shapes = [#tf_type.shape<>, #tf_type.shape<>], device = "", num_sparse = 2 : i64, ragge…
45 // CHECK-NEXT: key: "num_sparse"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | derived_attrs.mlir | 16 …{ device = "/device:CPU:0", num_sparse = 0 : i64, dense_shapes = [#tf_type.shape<>], result_segmen…
|
| D | fallback.mlir | 8 // CHECK-SAME: num_sparse = 2 : i64, ragged_split_types = [], ragged_value_types = []
35 // CHECK-SAME: num_sparse = 2 : i64
43 … device = "/device:CPU:0", num_sparse = 2 : i64, ragged_split_types = [], ragged_value_types = [],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/analysis/ |
| D | compatibility_analysis.mlir | 62 …{dense_shapes = [#tf_type.shape<>, #tf_type.shape<>], num_sparse = 0 : i64, result_segment_sizes =…
|
| /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 | 732 int num_sparse = 0; in PrepareParseExample() local
738 num_sparse = value.AsInt32(); in PrepareParseExample()
744 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 | 3542 …dense_default_1) {dense_shapes = [#tf_type.shape<>, #tf_type.shape<>], num_sparse = 0 : i64, resul…
3551 …dense_default_1) {dense_shapes = [#tf_type.shape<>, #tf_type.shape<>], num_sparse = 0 : i64, resul…
3560 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 2 : i64, resul…
3568 …// expected-error @+1 {{attribute 'num_sparse' should be the same as the length of attribute 'spar…
3569 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 3 : i64, resul…
3578 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 2 : i64, resul…
3587 …sparse_keys, %empty_str_vector, %empty_str_vector) {dense_shapes = [], num_sparse = 2 : i64, resul…
3596 …empty_str_vector, %empty_str_vector, %ragged_keys) {dense_shapes = [], num_sparse = 0 : i64, resul…
3605 …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 | 4327 num_sparse = pfor_input.get_attr("num_sparse")
4342 num_sparse=num_sparse,
|