From 57889aaedcb5285821bdee3362b8767bd6631ebb Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 21 Oct 2026 23:33:42 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 1004523690 --- tensorflow/core/tpu/ops/BUILD | 3 +++ tensorflow/core/tpu/ops/tpu_embedding_ops.cc | 2 ++ tensorflow/core/tpu/ops/tpu_embedding_shape_util.cc | 1 + tensorflow/core/tpu/ops/tpu_embedding_shape_util.h | 1 + 4 files changed, 7 insertions(+) diff --git a/tensorflow/core/tpu/ops/BUILD b/tensorflow/core/tpu/ops/BUILD index df36de4c3ca..fcedbba7e24 100644 --- a/tensorflow/core/tpu/ops/BUILD +++ b/tensorflow/core/tpu/ops/BUILD @@ -154,6 +154,7 @@ cc_library( "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@xla//xla:util", + "@xla//xla/tpu/protobuf:tpu_embedding_configuration_proto_cc", ], ) @@ -180,6 +181,8 @@ cc_library( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", "@xla//xla:status_macros", + "@xla//xla/tpu/protobuf:optimization_parameters_proto_cc", + "@xla//xla/tpu/protobuf:tpu_embedding_configuration_proto_cc", "@xla//xla/tsl/platform:errors", "@xla//xla/tsl/platform:logging", ], diff --git a/tensorflow/core/tpu/ops/tpu_embedding_ops.cc b/tensorflow/core/tpu/ops/tpu_embedding_ops.cc index 5a7bd72f9fd..0d68af42098 100644 --- a/tensorflow/core/tpu/ops/tpu_embedding_ops.cc +++ b/tensorflow/core/tpu/ops/tpu_embedding_ops.cc @@ -25,6 +25,8 @@ limitations under the License. #include "absl/strings/str_cat.h" #include "absl/strings/str_format.h" #include "xla/status_macros.h" +#include "xla/tpu/protobuf/optimization_parameters.pb.h" +#include "xla/tpu/protobuf/tpu_embedding_configuration.pb.h" #include "xla/tsl/platform/errors.h" #include "xla/tsl/platform/logging.h" // IWYU pragma: keep #include "tensorflow/core/framework/attr_value.pb.h" diff --git a/tensorflow/core/tpu/ops/tpu_embedding_shape_util.cc b/tensorflow/core/tpu/ops/tpu_embedding_shape_util.cc index b55f0589cf9..58d3afdecb0 100644 --- a/tensorflow/core/tpu/ops/tpu_embedding_shape_util.cc +++ b/tensorflow/core/tpu/ops/tpu_embedding_shape_util.cc @@ -22,6 +22,7 @@ limitations under the License. #include "absl/status/statusor.h" #include "absl/strings/str_cat.h" #include "absl/types/span.h" +#include "xla/tpu/protobuf/tpu_embedding_configuration.pb.h" #include "xla/tsl/platform/errors.h" #include "xla/tsl/platform/statusor.h" #include "xla/util.h" diff --git a/tensorflow/core/tpu/ops/tpu_embedding_shape_util.h b/tensorflow/core/tpu/ops/tpu_embedding_shape_util.h index 1d1e91382d2..abd2e3ba430 100644 --- a/tensorflow/core/tpu/ops/tpu_embedding_shape_util.h +++ b/tensorflow/core/tpu/ops/tpu_embedding_shape_util.h @@ -21,6 +21,7 @@ limitations under the License. #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/types/span.h" +#include "xla/tpu/protobuf/tpu_embedding_configuration.pb.h" #include "tensorflow/core/framework/tensor_shape.pb.h" #include "tensorflow/core/platform/status.h" #include "tensorflow/core/protobuf/tpu/tpu_embedding_configuration.pb.h"