Automated Code Change

PiperOrigin-RevId: 792021594

Author: tensorflower-gardener

third_party/xla/xla/python/transfer/BUILD

@@ -260,6 +260,7 @@ cc_library(
         ":streaming_ifrt",
         "//xla:shape_util",
         "//xla:util",
+        "//xla:xla_data_proto_cc",
         "//xla/pjrt:pjrt_client",
         "//xla/pjrt/distributed:key_value_store_interface",
         "//xla/python/ifrt",

third_party/xla/xla/python/transfer/pjrt_transfer_server.cc

@@ -56,6 +56,7 @@ limitations under the License.
 #include "xla/tsl/platform/errors.h"
 #include "xla/tsl/platform/statusor.h"
 #include "xla/util.h"
+#include "xla/xla_data.pb.h"
 #include "tsl/platform/casts.h"
 
 namespace xla {