Searched defs:dense_shapes (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/fuzzing/
H A Dexample_proto_fast_parsing_fuzz.cc43 std::vector<PartialTensorShape> dense_shapes; variable
44 dense_shapes.push_back(PartialTensorShape());
48 sparse_types, dense_shapes);
/external/tensorflow/tensorflow/core/example/
H A Dexample_parser_configuration.cc64 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/kernels/
H A Dexample_parsing_ops_test.cc155 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...]
/external/tensorflow/tensorflow/core/util/
H A Dexample_proto_helper.h161 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes));
164 for (int i = 0; i < dense_shapes.size(); ++i) {
166 if (dense_shapes[i].dims() == -1) {
169 for (int d = 1; d < dense_shapes[i].dims(); ++d) {
170 if (dense_shapes[i].dim_size(d) == -1) {
177 "dense_shapes[", i,
179 dense_shapes[i].DebugString());
182 if (dense_shapes[i].dims() > 0 && dense_shapes[
200 std::vector<PartialTensorShape> dense_shapes; member in class:tensorflow::ParseExampleAttrs
266 std::vector<PartialTensorShape> dense_shapes; member in class:tensorflow::ParseSingleExampleAttrs
[all...]

Completed in 114 milliseconds