Automated Code Change

PiperOrigin-RevId: 651983211

Author: tensorflower-gardener

third_party/xla/xla/python/ifrt_proxy/common/BUILD

@@ -139,6 +139,7 @@ cc_library(
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
+        "@com_google_absl//absl/types:span",
         "@local_tsl//tsl/platform:statusor",
     ],
 )

third_party/xla/xla/python/ifrt_proxy/common/array_util.h

@@ -22,6 +22,7 @@
 
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
+#include "absl/types/span.h"
 #include "xla/python/ifrt/dtype.h"
 #include "xla/python/ifrt/shape.h"
 

third_party/xla/xla/python/ifrt_proxy/common/proto_util.cc

@@ -18,6 +18,7 @@
 #include <memory>
 
 #include "absl/status/status.h"
+#include "xla/python/ifrt_proxy/common/ifrt_service.pb.h"
 #include "tsl/platform/status_to_from_proto.h"
 
 namespace xla {

third_party/xla/xla/python/ifrt_proxy/common/types.cc

@@ -26,6 +26,7 @@
 #include "absl/types/span.h"
 #include "xla/pjrt/pjrt_common.h"
 #include "xla/python/ifrt/array.h"
+#include "xla/python/ifrt_proxy/common/ifrt_service.pb.h"
 #include "xla/python/ifrt_proxy/common/types.pb.h"
 
 namespace xla {