Automated Code Change

PiperOrigin-RevId: 643541625
tensorflow · Jun 19, 2024 · 998238c · 998238c
1 parent c625df7
commit 998238c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tensorflow/core/kernels/scatter_nd_op_test.cc b/tensorflow/core/kernels/scatter_nd_op_test.cc
@@ -363,7 +363,7 @@ TEST_F(ScatterNdOpConstructionTest, Error_BadIndicesPolicyInvalid) {
                    .Input(FakeInput(DT_INT32))
                    .Attr("bad_indices_policy", "AN_UNRECOGNIZED_POLICY")
                    .Finalize(node_def()));
-  EXPECT_NE(InitOp(), OkStatus());
+  EXPECT_NE(InitOp(), absl::OkStatus());
 }
 
 class ScatterNdUpdateBM : public ScatterNdUpdateOpTest {

diff --git a/third_party/xla/xla/service/gpu/triton_support_test.cc b/third_party/xla/xla/service/gpu/triton_support_test.cc
@@ -51,8 +51,8 @@ auto AllXlaDataTypes() {
   std::vector<xla::PrimitiveType> xla_data_types;
   std::vector<xla::PrimitiveType> to_filter_out = {PRIMITIVE_TYPE_INVALID,
                                                    TUPLE, OPAQUE_TYPE, TOKEN};
-  const proto2::EnumDescriptor* xla_type_descriptor =
-      proto2::GetEnumDescriptor<xla::PrimitiveType>();
+  const tsl::protobuf::EnumDescriptor* xla_type_descriptor =
+      tsl::protobuf::GetEnumDescriptor<xla::PrimitiveType>();
   for (int enum_ix = 0; enum_ix < xla_type_descriptor->value_count();
        ++enum_ix) {
     xla::PrimitiveType xla_type = static_cast<xla::PrimitiveType>(