diff options
| author |  Eugene Brevdo <ebrevdo@google.com> | 2017-01-31 09:21:25 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-31 09:51:32 -0800 |
| commit | a32dcc0c832929428067e93b1c05b0ef69096c0a (patch) | |
| tree | 5f6bf9e136c2aee458de637e1f1d98c9352e1730 /tensorflow/core/util/example_proto_fast_parsing.h | |
| parent | 97c8485add110a3133e667f986c84b6a190e06e2 (diff) | |
parse_example_op dense_shape attribute TensorShape -> PartialTensorShape.
This is functionally a no-op and should have no performance implications. In
followup CLs, we will start taking advantage of this flexibility.
Change: 146127565
Diffstat (limited to 'tensorflow/core/util/example_proto_fast_parsing.h')
| -rw-r--r-- | tensorflow/core/util/example_proto_fast_parsing.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/util/example_proto_fast_parsing.h b/tensorflow/core/util/example_proto_fast_parsing.h index 6ed9d57838..4878199802 100644 --- a/tensorflow/core/util/example_proto_fast_parsing.h +++ b/tensorflow/core/util/example_proto_fast_parsing.h @@ -24,6 +24,7 @@ limitations under the License. #include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/op_kernel.h" +#include "tensorflow/core/framework/partial_tensor_shape.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/lib/gtl/array_slice.h" @@ -45,8 +46,9 @@ struct FastParseExampleConfig { // These 2 fields correspond exactly to dense_shapes and dense_defaults in // ParseExample op. // Documentation is avaliable in: tensorflow/core/ops/parsing_ops.cc - TensorShape shape; + PartialTensorShape shape; Tensor default_value; + std::size_t elements_per_stride; }; struct Sparse { |