Commit b0a49cd0 authored by Eugene Brevdo's avatar Eugene Brevdo Committed by TensorFlower Gardener
Browse files

TensorArray changes: respect infer_shape argument inside the TensorArray object. 

This adds a new attr to the TensorArrayV3 construction op: identical_element_shapes (default False).

If True, then shape for all elements is inferred at runtime when any single element is written.

The Python TensorArray constructor's "infer_shape" argument is piped through to this attribute.  Since it is true by default, this enables runtime element consistency checking on top of the existing graph build time static shape checking.

PiperOrigin-RevId: 175862771
parent 51e8b01c
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment