tree 69ed098ea7d34cd9d51e7a80e898fd3f7e20eedb
parent f3cb68f0ac458281e5b06cd0066607d112adb0e2
author Smit Hinsu <hinsu@google.com> 1592619514 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1592620111 -0700

Rewrite masks while applying ellipsis mask to tf.StridedSlice op

begin_mask and end_mask needs to be adjusted while simplifying StridedSlice op by removing ellipsis_mask. new_axis_mask is already removed before this and we don't yet support shrink_axis_mask.

PiperOrigin-RevId: 317419960
Change-Id: Ie4a5f404f95f5b909065311a54cbbed64d4ccf4b