tree b0180b63d79c5fde3988b2215e5139b0f77132f9
parent bfc75faae77274ff9a0d542a3f271d5424f54a61
author Andy Ly <lyandy@google.com> 1611080165 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1611080780 -0800

Update DistributedTPURewritePass to populate OpMetadata in OpSharding.

Arguments and results to a TPU computation will now have OpMetadata containing the source of the sharding attribute (TF op type and name).

PiperOrigin-RevId: 352596417
Change-Id: I75a7114f039b9938ac1479d61ffd37a3ed0f843d