tree 49377cc1c22ce9383c46c60da74bd0e3c6402cbb
parent 37406182181eaf0356fa832b26a3a20e8f2f78a1
author A. Unique TensorFlower <gardener@tensorflow.org> 1613239286 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1613239671 -0800

We just need to copy the tiling info of result shape in layout assignment instead
the whole shape since the shapes already match.

PiperOrigin-RevId: 357367492
Change-Id: I05be8ebe78265f6d5d4de00c093b8008631275b2