Searched refs:num_dense (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/util/ |
| D | example_parser_configuration.py | 51 num_dense = parse_example_op.get_attr("Ndense")
52 total_features = num_dense + num_sparse
63 if len(dense_types) != num_dense:
66 (len(dense_types), num_dense))
68 if len(dense_shapes) != num_dense:
71 (len(dense_shapes), num_dense))
77 if len(fetch_list) != (total_features + num_dense):
80 (len(fetch_list), (total_features + num_dense)))
91 dense_def_start = dense_keys_start + num_dense
100 for i in range(num_dense):
[all …]
|
| /external/tensorflow/tensorflow/core/example/ |
| D | example_parser_configuration.cc | 59 auto num_dense = attr_map.at("Ndense").i(); in ExtractExampleParserConfiguration() local
60 fixed_len_features->resize(num_dense); in ExtractExampleParserConfiguration()
68 if (tdense.list().type_size() != num_dense) { in ExtractExampleParserConfiguration()
71 " elements != Ndense attr: ", num_dense); in ExtractExampleParserConfiguration()
74 if (dense_shapes.list().shape_size() != num_dense) { in ExtractExampleParserConfiguration()
77 " elements != Ndense attr: ", num_dense); in ExtractExampleParserConfiguration()
113 int dense_defaults_start = dense_keys_start + num_dense; in ExtractExampleParserConfiguration()
121 for (int i = 0; i < num_dense; ++i) { in ExtractExampleParserConfiguration()
155 for (int i = 0; i < num_dense; ++i) { in ExtractExampleParserConfiguration()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops_test.cc | 88 auto set_outputs = [&op](int num_sparse, int num_dense, in TEST()
100 .Input(NodeOutList(num_dense, string_in)) in TEST()
101 .Input(NodeOutList(num_dense, string_in)) in TEST()
104 .Attr("dense_shapes", MakeDenseShapes(num_dense, add_extra_shape, in TEST()
305 auto set_outputs = [&op](int num_sparse, int num_dense, int num_ragged, in TEST()
319 .Input(NodeOutList(num_dense, string_in)) // dense_defaults in TEST()
324 .Attr("dense_shapes", MakeDenseShapes(num_dense, add_extra_shape, in TEST()
|
| D | parsing_ops.cc | 88 void AddDenseLengthsShapes(int num_dense, const ShapeHandle& shape, in AddDenseLengthsShapes() argument
90 for (int i = 0; i < num_dense; ++i) { in AddDenseLengthsShapes()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParseExampleV2.pbtxt | 92 A list of `num_dense` shapes; the shapes of data in each Feature
93 given in dense_keys (where `num_dense = dense_keys.size()`).
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.cc | 414 num_dense = dense_types.size(); in FinishInit()
423 if (static_cast<size_t>(num_dense) != dense_types.size()) { in FinishInit()
426 if (static_cast<size_t>(num_dense) != dense_shapes.size()) { in FinishInit()
437 if (num_dense > std::numeric_limits<int32>::max()) { in FinishInit()
|
| D | example_proto_helper.h | 174 TF_RETURN_IF_ERROR(ctx->GetAttr("Ndense", &num_dense));
190 int64 num_dense; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 151 if (dense_keys_t.size() != attrs_.num_dense) { in CheckInputShapes()
154 dense_keys_t.size(), " vs. ", attrs_.num_dense); in CheckInputShapes()
168 if (dense_defaults.size() != attrs_.num_dense) { in CheckInputShapes()
171 dense_defaults.size(), " vs. ", attrs_.num_dense); in CheckInputShapes()
174 for (int d = 0; d < static_cast<int>(attrs_.num_dense); ++d) { in CheckInputShapes()
214 config.dense.reserve(attrs_.num_dense); in MakeConfig()
215 for (int d = 0; d < attrs_.num_dense; ++d) { in MakeConfig()
265 for (int d = 0; d < attrs_.num_dense; ++d) { in WriteOutput()
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 731 int num_dense = 0; in PrepareParseExample() local
739 num_dense = value.AsInt32(); in PrepareParseExample()
743 data->dense_size = num_dense; in PrepareParseExample()
|