| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ParseSingleExample.pbtxt | 17 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,
|
| D | api_def_ParseExample.pbtxt | 42 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
|
| D | api_def_ParseExampleV2.pbtxt | 50 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/ |
| D | example_parser_configuration.py | 54 dense_shapes = parse_example_op.get_attr("dense_shapes")
66 if len(dense_shapes) != num_dense:
69 (len(dense_shapes), num_dense))
110 tensor_shape.TensorShape(dense_shapes[i]).as_proto())
142 dense_shapes = parse_example_op.get_attr("dense_shapes")
180 tensor_shape.TensorShape(dense_shapes[i]).as_proto())
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops_test.cc | 164 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/kernels/fuzzing/ |
| D | example_proto_fast_parsing_fuzz.cc | 43 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/util/ |
| D | example_proto_helper.cc | 426 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 …]
|
| D | example_proto_helper.h | 157 Status GetDenseShapes(const std::vector<PartialTensorShape>& dense_shapes,
169 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes));
171 GetDenseShapes(dense_shapes, &variable_length, &elements_per_stride));
197 std::vector<PartialTensorShape> dense_shapes; member
215 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes)); in Init()
226 GetDenseShapes(dense_shapes, &variable_length, &elements_per_stride)); in Init()
234 std::vector<PartialTensorShape> dense_shapes; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_config.py | 396 dense_shapes=None, argument
408 dense_shapes = ([[]] *
409 len(dense_keys) if dense_shapes is None else dense_shapes)
419 self.dense_shapes = [tensor_shape.as_shape(s) for s in dense_shapes]
458 return [shape.as_proto() for shape in self.dense_shapes]
468 for k, s, t in zip(self.dense_keys, self.dense_shapes, self.dense_types)
582 self.dense_shapes.append(tensor_shape.as_shape(feature.shape))
596 self.dense_shapes.append(tensor_shape.as_shape(feature.shape))
630 if len(self.dense_shapes) != len(self.dense_keys):
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 655 std::vector<tf::TensorShape> dense_shapes; member
720 auto dense_shapes = nodedef.attr().at("dense_shapes").list(); in PrepareParseExample() local
721 if (data->dense_shapes.empty()) { in PrepareParseExample()
722 for (int i = 0; i < dense_shapes.shape_size(); ++i) { in PrepareParseExample()
723 data->dense_shapes.push_back(dense_shapes.shape(i)); in PrepareParseExample()
756 data->dense_shapes.reserve(data->dense_size); in PrepareParseExample()
760 const bool missing_shape_info = data->dense_shapes.empty(); in PrepareParseExample()
766 data->dense_shapes.push_back(TfLiteToTfShape(output_size)); in PrepareParseExample()
769 const int original_size = data->dense_shapes[i].dims() > 0 in PrepareParseExample()
770 ? data->dense_shapes[i].dim_size(0) in PrepareParseExample()
[all …]
|
| D | parse_example_test.cc | 62 key: "dense_shapes"
90 key: "dense_shapes"
118 key: "dense_shapes"
146 key: "dense_shapes"
175 key: "dense_shapes"
|
| /external/tensorflow/tensorflow/core/example/ |
| D | example_parser_configuration.cc | 64 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/core/ops/compat/ops_history_v2/ |
| D | ExperimentalParseExampleDataset.pbtxt | 54 name: "dense_shapes"
124 name: "dense_shapes"
208 name: "dense_shapes"
305 name: "dense_shapes"
|
| D | ParseExampleDatasetV2.pbtxt | 54 name: "dense_shapes"
178 name: "dense_shapes"
315 name: "dense_shapes"
|
| D | ParseExampleDataset.pbtxt | 54 name: "dense_shapes"
178 name: "dense_shapes"
315 name: "dense_shapes"
|
| D | ParseSingleExample.pbtxt | 69 name: "dense_shapes"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | derived_attrs.mlir | 15 // CHECK-SAME: dense_shapes = [#corert.shape<>]
16 …{ device = "/device:CPU:0", num_sparse = 0 : i64, dense_shapes = [#tf_type.shape<>], result_segmen…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops.cc | 32 Status AddDenseOutputShapes(const std::vector<TensorShapeType>& dense_shapes, in AddDenseOutputShapes() argument
35 for (const auto& dense_shape : dense_shapes) { in AddDenseOutputShapes()
174 AddDenseOutputShapes(attrs.dense_shapes, input, c, &output_idx)); in __anon5cc8b68c0402()
222 AddDenseOutputShapes(attrs.dense_shapes, input, c, &output_idx)); in __anon5cc8b68c0502()
252 AddDenseOutputShapes(attrs.dense_shapes, input, c, &output_idx)); in __anon5cc8b68c0602()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ExperimentalParseExampleDataset.pbtxt | 54 name: "dense_shapes"
124 name: "dense_shapes"
|
| D | ParseExampleDataset.pbtxt | 54 name: "dense_shapes"
131 name: "dense_shapes"
|
| D | ParseSingleExample.pbtxt | 69 name: "dense_shapes"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | parse_example.mlir | 30 // CHECK: key: "dense_shapes"
56 …raps "tf.ParseExample"(%outputs_4, %outputs, %outputs_0, %outputs_2) {dense_shapes = [#tf_type.sha…
|
| D | parse_example_v2.mlir | 14 …puts, %outputs_0) {Tdense = ["tfdtype$DT_FLOAT", "tfdtype$DT_FLOAT"], dense_shapes = [#tf_type.sha…
34 // CHECK: key: "dense_shapes"
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/graphdef_to_mlir/ |
| D | invalid_empty_func_attr_name.pbtxt | 9 key: "dense_shapes"
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | handle_sparsetensors.cc | 60 mlir::RankedTensorType dense_shapes; member
165 main_func.front().addArgument(component.dense_shapes, main_func.getLoc()); in UpdateFunctionWithSparseTensorComponents()
|