From 998238c5f5c96fa53213c0ff2d4df1ed48649709 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 14 Jun 2024 22:35:14 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 643541625 --- tensorflow/core/kernels/scatter_nd_op_test.cc | 2 +- third_party/xla/xla/service/gpu/triton_support_test.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/core/kernels/scatter_nd_op_test.cc b/tensorflow/core/kernels/scatter_nd_op_test.cc index eb5f42d012ad8c..fd679c35347798 100644 --- 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 index fb818d6efab617..6c690bad4c47a9 100644 --- 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_data_types; std::vector to_filter_out = {PRIMITIVE_TYPE_INVALID, TUPLE, OPAQUE_TYPE, TOKEN}; - const proto2::EnumDescriptor* xla_type_descriptor = - proto2::GetEnumDescriptor(); + const tsl::protobuf::EnumDescriptor* xla_type_descriptor = + tsl::protobuf::GetEnumDescriptor(); for (int enum_ix = 0; enum_ix < xla_type_descriptor->value_count(); ++enum_ix) { xla::PrimitiveType xla_type = static_cast(