Home
last modified time | relevance | path

Searched refs:dense_shapes (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ParseSingleExample.pbtxt17 If dense_defaults[j] is not empty, and dense_shapes[j] is fully defined,
18 then the shape of dense_defaults[j] must match that of dense_shapes[j].
19 If dense_shapes[j] has an undefined major dimension (variable strides dense
64 name: "dense_shapes"
69 always equal dense_shapes[j].NumEntries(). If dense_shapes[j] ==
71 will be (D0, D1, ..., DN): In the case dense_shapes[j] = (-1, D1,
Dapi_def_ParseExample.pbtxt42 If dense_defaults[j] is not empty, and dense_shapes[j] is fully defined,
43 then the shape of dense_defaults[j] must match that of dense_shapes[j].
44 If dense_shapes[j] has an undefined major dimension (variable strides dense
59 name: "dense_shapes"
64 must always equal dense_shapes[j].NumEntries().
65 If dense_shapes[j] == (D0, D1, ..., DN) then the shape of output
68 This works for dense_shapes[j] = (-1, D1, ..., DN). In this case
Dapi_def_ParseExampleV2.pbtxt50 If dense_defaults[j] is not empty, and dense_shapes[j] is fully defined,
51 then the shape of dense_defaults[j] must match that of dense_shapes[j].
52 If dense_shapes[j] has an undefined major dimension (variable strides dense
90 name: "dense_shapes"
95 must always equal dense_shapes[j].NumEntries().
96 If dense_shapes[j] == (D0, D1, ..., DN) then the shape of output
99 This works for dense_shapes[j] = (-1, D1, ..., DN). In this case
/external/tensorflow/tensorflow/python/util/
Dexample_parser_configuration.py56 dense_shapes = parse_example_op.get_attr("dense_shapes")
68 if len(dense_shapes) != num_dense:
71 (len(dense_shapes), num_dense))
112 tensor_shape.TensorShape(dense_shapes[i]).as_proto())
144 dense_shapes = parse_example_op.get_attr("dense_shapes")
182 tensor_shape.TensorShape(dense_shapes[i]).as_proto())
/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dexample_proto_fast_parsing_fuzz.cc43 std::vector<PartialTensorShape> dense_shapes; in BuildGraph() local
44 dense_shapes.push_back(PartialTensorShape()); in BuildGraph()
48 dense_shapes); in BuildGraph()
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops_test.cc164 std::vector<PartialTensorShape> dense_shapes; in ParseExample() local
174 dense_shapes.push_back(PartialTensorShape({feature_size})); in ParseExample()
180 dense_shapes.push_back(PartialTensorShape({-1})); in ParseExample()
197 .Attr("dense_shapes", dense_shapes) in ParseExample()
227 std::vector<PartialTensorShape> dense_shapes; in ParseExampleV2() local
237 dense_shapes.push_back(PartialTensorShape({feature_size})); in ParseExampleV2()
242 dense_shapes.push_back(PartialTensorShape({-1})); in ParseExampleV2()
274 .Attr("dense_shapes", dense_shapes) in ParseExampleV2()
294 std::vector<PartialTensorShape> dense_shapes; in ParseSingleExample() local
303 dense_shapes.push_back(PartialTensorShape({feature_size})); in ParseSingleExample()
[all …]
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.cc426 if (static_cast<size_t>(num_dense) != dense_shapes.size()) { in FinishInit()
465 if (dense_keys.size() != dense_shapes.size()) { in FinishInit()
633 Status GetDenseShapes(const std::vector<PartialTensorShape>& dense_shapes, in GetDenseShapes() argument
638 for (int i = 0; i < dense_shapes.size(); ++i) { in GetDenseShapes()
640 if (dense_shapes[i].dims() == -1) { in GetDenseShapes()
643 for (int d = 1; d < dense_shapes[i].dims(); ++d) { in GetDenseShapes()
644 if (dense_shapes[i].dim_size(d) == -1) { in GetDenseShapes()
653 dense_shapes[i].DebugString()); in GetDenseShapes()
656 if (dense_shapes[i].dims() > 0 && dense_shapes[i].dim_size(0) == -1) { in GetDenseShapes()
658 for (int d = 1; d < dense_shapes[i].dims(); ++d) { in GetDenseShapes()
[all …]
Dexample_proto_helper.h156 Status GetDenseShapes(const std::vector<PartialTensorShape>& dense_shapes,
168 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes));
170 GetDenseShapes(dense_shapes, &variable_length, &elements_per_stride));
196 std::vector<PartialTensorShape> dense_shapes; member
214 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes)); in Init()
225 GetDenseShapes(dense_shapes, &variable_length, &elements_per_stride)); in Init()
233 std::vector<PartialTensorShape> dense_shapes; member
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py397 dense_shapes=None, argument
409 dense_shapes = ([[]] *
410 len(dense_keys) if dense_shapes is None else dense_shapes)
420 self.dense_shapes = [tensor_shape.as_shape(s) for s in dense_shapes]
459 return [shape.as_proto() for shape in self.dense_shapes]
469 for k, s, t in zip(self.dense_keys, self.dense_shapes, self.dense_types)
578 self.dense_shapes.append(tensor_shape.as_shape(feature.shape))
590 self.dense_shapes.append(tensor_shape.as_shape(feature.shape))
624 if len(self.dense_shapes) != len(self.dense_keys):
627 (len(self.dense_shapes), len(self.dense_keys)))
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc64 auto dense_shapes = attr_map.at("dense_shapes"); in ExtractExampleParserConfiguration() local
74 if (dense_shapes.list().shape_size() != num_dense) { in ExtractExampleParserConfiguration()
76 dense_shapes.list().shape_size(), in ExtractExampleParserConfiguration()
89 (*fixed_len_features)[i].shape = TensorShape(dense_shapes.list().shape(i)); in ExtractExampleParserConfiguration()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc654 std::vector<tf::TensorShape> dense_shapes; member
719 auto dense_shapes = nodedef.attr().at("dense_shapes").list(); in PrepareParseExample() local
720 for (int i = 0; i < dense_shapes.shape_size(); ++i) { in PrepareParseExample()
721 data->dense_shapes.push_back(dense_shapes.shape(i)); in PrepareParseExample()
753 data->dense_shapes.reserve(data->dense_size); in PrepareParseExample()
757 const bool missing_shape_info = data->dense_shapes.empty(); in PrepareParseExample()
764 data->dense_shapes.push_back(TfLiteToTfShape(output_size)); in PrepareParseExample()
836 const auto& dense_shapes = data->dense_shapes; in EvalParseExample() local
855 dense_shapes[i].dims() ? dense_shapes[i].num_elements() : 1; in EvalParseExample()
859 k, tf::DataTypeToEnum<tf::int64>::value, dense_shapes[i], in EvalParseExample()
[all …]
Dparse_example_test.cc60 key: "dense_shapes"
88 key: "dense_shapes"
116 key: "dense_shapes"
144 key: "dense_shapes"
/external/tensorflow/tensorflow/core/ops/
Dparsing_ops.cc32 Status AddDenseOutputShapes(const std::vector<TensorShapeType>& dense_shapes, in AddDenseOutputShapes() argument
35 for (const auto& dense_shape : dense_shapes) { in AddDenseOutputShapes()
172 AddDenseOutputShapes(attrs.dense_shapes, input, c, &output_idx)); in __anonf8f6e66e0402()
220 AddDenseOutputShapes(attrs.dense_shapes, input, c, &output_idx)); in __anonf8f6e66e0502()
250 AddDenseOutputShapes(attrs.dense_shapes, input, c, &output_idx)); in __anonf8f6e66e0602()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DExperimentalParseExampleDataset.pbtxt54 name: "dense_shapes"
124 name: "dense_shapes"
DParseExampleDataset.pbtxt54 name: "dense_shapes"
131 name: "dense_shapes"
DParseSingleExample.pbtxt69 name: "dense_shapes"
DParseExample.pbtxt78 name: "dense_shapes"
DParseExampleDatasetV2.pbtxt54 name: "dense_shapes"
DParseExampleV2.pbtxt106 name: "dense_shapes"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DExperimentalParseExampleDataset.pbtxt54 name: "dense_shapes"
124 name: "dense_shapes"
DParseSingleExample.pbtxt69 name: "dense_shapes"
DParseExample.pbtxt78 name: "dense_shapes"
DParseExampleDatasetV2.pbtxt54 name: "dense_shapes"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dparse_example.mlir30 // CHECK: key: "dense_shapes"
56 …raps "tf.ParseExample"(%outputs_4, %outputs, %outputs_0, %outputs_2) {dense_shapes = [#tf.shape<1>…
Dparse_example_v2.mlir14 …puts, %outputs_0) {Tdense = ["tfdtype$DT_FLOAT", "tfdtype$DT_FLOAT"], dense_shapes = [#tf.shape<>,…
34 // CHECK: key: "dense_shapes"

12