Introduce additional XLA TPU Ops to open source ยท tensorflow/tensorflow@3ea5fc7

PiperOrigin-RevId: 326343558 Change-Id: I47da1dc0c96cdf8223ccebef012e2a5088a857a4
github.com github.com