tree eecedc1b91f5bb5a68bebffc785ef0ac7ff6a92e
parent 1d602bea4876ff679d3b1ff51f7a091ee9924a52
author Derek Murray <mrry@google.com> 1582936085 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1582936271 -0800

Roll-forward of: [Grappler] Extend the RemoveStackStridedSliceSameAxis optimization to support Slice.

`tf.slice(tf.pack([x, y, z]), ...)` will always have rank one greater than `x,
y, z`. The previous version had a bug where it would incorrectly return the
original `x, y, z` tensor directly, if insufficient shape information were
available. The fix is to explicitly track whether the rank should be expanded or not, based on the slicing operation used.

PiperOrigin-RevId: 297961546
Change-Id: I80b10bd0cd39e783f8a9a41e9d0d5a3dcc250b82