| /external/tensorflow/tensorflow/core/util/ |
| example_proto_helper.h | 162 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes));
165 for (int i = 0; i < dense_shapes.size(); ++i) {
167 if (dense_shapes[i].dims() == -1) {
170 for (int d = 1; d < dense_shapes[i].dims(); ++d) {
171 if (dense_shapes[i].dim_size(d) == -1) {
178 "dense_shapes[", i,
180 dense_shapes[i].DebugString());
183 if (dense_shapes[i].dims() > 0 && dense_shapes[i].dim_size(0) == -1)
201 std::vector<PartialTensorShape> dense_shapes; member in class:tensorflow::ParseExampleAttrs
267 std::vector<PartialTensorShape> dense_shapes; member in class:tensorflow::ParseSingleExampleAttrs
[all...] |
| example_proto_helper.cc | 412 if (static_cast<size_t>(num_dense) != dense_shapes.size()) {
413 return errors::InvalidArgument("len(dense_keys) != len(dense_shapes)");
434 if (dense_keys.size() != dense_shapes.size()) {
435 return errors::InvalidArgument("len(dense_keys) != len(dense_shapes)");
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| example_proto_fast_parsing_fuzz.cc | 43 std::vector<PartialTensorShape> dense_shapes; variable
44 dense_shapes.push_back(PartialTensorShape());
48 dense_shapes);
|
| /external/tensorflow/tensorflow/python/util/ |
| example_parser_configuration.py | 46 dense_shapes = parse_example_op.get_attr("dense_shapes")
58 if len(dense_shapes) != num_dense:
59 raise ValueError("len(dense_shapes) attribute does not match "
61 (len(dense_shapes), num_dense))
102 tensor_shape.TensorShape(dense_shapes[i]).as_proto())
|
| /external/tensorflow/tensorflow/python/ops/ |
| parsing_ops.py | 199 `dense_defaults`, `dense_shapes`.
214 dense_shapes = []
279 dense_shapes.append(feature.shape)
292 dense_shapes.append(feature.shape)
302 dense_shapes)
801 dense_shapes) = _features_to_raw_params(
806 dense_types, dense_defaults, dense_shapes, name)
817 dense_shapes=None,
838 dense_shapes: A list of tuples with the same length as `dense_keys`.
855 dense_keys, dense_shapes, _) = _process_raw_parameters
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| parsing_ops.py | 43 dense_shapes) = parsing_ops._features_to_raw_params(
49 (_, dense_defaults_vec, sparse_keys, sparse_types, dense_keys, dense_shapes,
52 dense_types, dense_shapes)
58 self._dense_shapes = dense_shapes
|
| /external/tensorflow/tensorflow/core/kernels/ |
| example_parsing_ops_test.cc | 155 std::vector<PartialTensorShape> dense_shapes; local
165 dense_shapes.push_back(PartialTensorShape({feature_size}));
171 dense_shapes.push_back(PartialTensorShape({-1}));
188 .Attr("dense_shapes", dense_shapes)
207 std::vector<PartialTensorShape> dense_shapes; local
216 dense_shapes.push_back(PartialTensorShape({feature_size}));
222 dense_shapes.push_back(PartialTensorShape({-1}));
238 .Attr("dense_shapes", dense_shapes)
[all...] |
| example_parsing_ops.cc | 99 attrs_.dense_shapes[d].DebugString(),
108 attrs_.dense_shapes[d].IsCompatibleWith(def_value.shape()),
113 "] == ", attrs_.dense_shapes[d].DebugString()));
127 attrs_.dense_shapes[d], dense_defaults[d],
200 attrs_.dense_shapes[d].DebugString(),
209 attrs_.dense_shapes[d].IsCompatibleWith(def_value.shape()),
214 "] == ", attrs_.dense_shapes[d].DebugString()));
229 attrs_.dense_shapes[d], dense_defaults[d],
[all...] |
| /external/tensorflow/tensorflow/core/example/ |
| example_parser_configuration.cc | 64 auto dense_shapes = attr_map.at("dense_shapes"); local
74 if (dense_shapes.list().shape_size() != num_dense) {
75 return errors::InvalidArgument("Node attr dense_shapes has ",
76 dense_shapes.list().shape_size(),
89 (*fixed_len_features)[i].shape = TensorShape(dense_shapes.list().shape(i));
|
| /external/tensorflow/tensorflow/core/ops/ |
| parsing_ops.cc | 63 .Attr("dense_shapes: list(shape) >= 0")
85 // Output dense_shapes.
89 c->MakeShapeFromPartialTensorShape(attrs.dense_shapes[i], &dense));
108 .Attr("dense_shapes: list(shape) >= 0")
128 // Output dense_shapes.
132 c->MakeShapeFromPartialTensorShape(attrs.dense_shapes[i], &dense));
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| parse_example_dataset_op.cc | 38 OP_REQUIRES_OK(ctx, ctx->GetAttr("dense_shapes", &dense_shapes_));
55 "dense_shapes[", i,
163 const std::vector<PartialTensorShape>& dense_shapes,
176 dense_shapes_(dense_shapes),
253 {"dense_shapes", dense_shapes_attr},
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sparse_ops_test.py | 802 dense_shapes = [(10, 10, 1), (5, 5), (1,), (1, 7)]
[all...] |
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| control_flow_ops_test.py | [all...] |
| pfor.py | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_experimental.cc | 754 key: "dense_shapes"
[all...] |
