diff options
author | Peter Hawkins <phawkins@google.com> | 2017-06-29 08:10:45 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-29 08:15:28 -0700 |
commit | 0c9b8974b49c147942af5888e73e325d42574db8 (patch) | |
tree | 665463fa25fab0783e23960e08bfe96684d3024f /tensorflow/core/kernels/constant_op.h | |
parent | 56d54aa44b81bf9c51eec2ffa0a460e597aced16 (diff) |
Fix shape error when a Placeholder operator with a partial shape is executed and not fed. The PlaceholderOp constructor was attempting to convert its shape attribute to a TensorShape, which requires that the shape is fully defined. This failed with a misleading error message for a partial shape. Instead, use a PartialTensorShape.
PiperOrigin-RevId: 160526882
Diffstat (limited to 'tensorflow/core/kernels/constant_op.h')
-rw-r--r-- | tensorflow/core/kernels/constant_op.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/constant_op.h b/tensorflow/core/kernels/constant_op.h index 8531a7a4b4..b98153e347 100644 --- a/tensorflow/core/kernels/constant_op.h +++ b/tensorflow/core/kernels/constant_op.h @@ -56,7 +56,7 @@ class PlaceholderOp : public OpKernel { void Compute(OpKernelContext* ctx) override; private: - TensorShape expected_shape_; + PartialTensorShape expected_shape_; }; } // namespace tensorflow |