Automated rollback of commit 31c0857

PiperOrigin-RevId: 213764810

tensorflow/compiler/aot/tests/BUILD

@@ -25,7 +25,6 @@ test_suite(
         ":test_graph_tfmatmul_test",
         ":test_graph_tfmatmulandadd_test",
         ":test_graph_tfsplits_test",
-        ":test_graph_tftop_k_test",
         ":tfcompile_test",
     ],
 )
@@ -43,7 +42,6 @@ py_binary(
         "//tensorflow/python:control_flow_ops",
         "//tensorflow/python:framework_for_generated_wrappers",
         "//tensorflow/python:math_ops",
-        "//tensorflow/python:nn_ops",
         "//tensorflow/python:platform",
         "//tensorflow/python:session",
         "//tensorflow/python:training",
@@ -68,7 +66,6 @@ genrule(
         "test_graph_tfmatmul.pb",
         "test_graph_tfmatmulandadd.pb",
         "test_graph_tfsplits.pb",
-        "test_graph_tftop_k.pb",
     ],
     # Set CUDA_VISIBLE_DEVICES='' to prevent the code we launch from using any
     # GPUs which might be present.  This is important because builds may run
@@ -211,17 +208,6 @@ tf_library(
     ],
 )
 
-tf_library(
-    name = "test_graph_tftop_k",
-    testonly = 1,
-    config = "test_graph_tftop_k.config.pbtxt",
-    cpp_class = "TopKComp",
-    graph = "test_graph_tftop_k.pb",
-    tags = [
-        "manual",
-    ],
-)
-
 tf_cc_test(
     name = "tfcompile_test",
     srcs = ["tfcompile_test.cc"],
@@ -240,7 +226,6 @@ tf_cc_test(
         ":test_graph_tfmatmulandadd",
         ":test_graph_tfmatmulandadd_with_profiling",
         ":test_graph_tfsplits",
-        ":test_graph_tftop_k",
         "//tensorflow/compiler/xla:shape_util",
         "//tensorflow/compiler/xla:test",
         "//tensorflow/compiler/xla:xla_data_proto",

tensorflow/compiler/aot/tests/make_test_graphs.py

@@ -31,7 +31,6 @@
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import control_flow_ops
 from tensorflow.python.ops import math_ops
-from tensorflow.python.ops import nn_ops
 from tensorflow.python.ops import variables
 from tensorflow.python.platform import app
 from tensorflow.python.training import saver as saver_lib
@@ -143,12 +142,6 @@ def tfsplits(_):
   array_ops.identity(y, name='result')
 
 
-def tftop_k(_):
-  x = array_ops.placeholder(dtypes.int32, shape=[5], name='x')
-  output = nn_ops.top_k(x, 2, name='values')
-  array_ops.identity(output[1], name='indices')
-
-
 def write_graph(build_graph, out_dir):
   """Build a graph using build_graph and write it out."""
   g = ops.Graph()
@@ -170,7 +163,6 @@ def main(_):
   write_graph(tfmatmul, FLAGS.out_dir)
   write_graph(tfmatmulandadd, FLAGS.out_dir)
   write_graph(tfsplits, FLAGS.out_dir)
-  write_graph(tftop_k, FLAGS.out_dir)
 
 
 if __name__ == '__main__':

tensorflow/compiler/aot/tests/tfcompile_test.cc

@@ -29,7 +29,6 @@ limitations under the License.
 #include "tensorflow/compiler/aot/tests/test_graph_tfmatmulandadd.h"
 #include "tensorflow/compiler/aot/tests/test_graph_tfmatmulandadd_with_profiling.h"
 #include "tensorflow/compiler/aot/tests/test_graph_tfsplits.h"
-#include "tensorflow/compiler/aot/tests/test_graph_tftop_k.h"
 #include "tensorflow/compiler/xla/service/hlo_profile_printer.h"
 #include "tensorflow/compiler/xla/shape_util.h"
 #include "tensorflow/compiler/xla/test.h"
@@ -449,30 +448,6 @@ TEST(TFCompileTest, Splits) {
   EXPECT_NEAR(expected[3], fn.result0(1, 1), 1e4);
 }
 
-TEST(TFCompileTest, TopK) {
-  Eigen::ThreadPool tp(1);
-  Eigen::ThreadPoolDevice device(&tp, tp.NumThreads());
-
-  TopKComp fn;
-
-  fn.set_thread_pool(&device);
-  // x = [4, 1, 4, 4, 3]
-  fn.arg0(0) = 4;
-  fn.arg0(1) = 1;
-  fn.arg0(2) = 4;
-  fn.arg0(3) = 4;
-  fn.arg0(4) = 3;
-
-  EXPECT_TRUE(fn.Run());
-  EXPECT_EQ(fn.error_msg(), "");
-  const int32 expected_values[] = {4, 4};
-  const int32 expected_indices[] = {0, 2};
-  EXPECT_EQ(expected_values[0], fn.result0(0));
-  EXPECT_EQ(expected_values[1], fn.result0(1));
-  EXPECT_EQ(expected_indices[0], fn.result1(0));
-  EXPECT_EQ(expected_indices[1], fn.result1(1));
-}
-
 TEST(TFCompileTest, AssertEqAndReturnDiff) {
   // Assert is converted into a no-op in XLA, so there is no failure even if the
   // two args are different.

tensorflow/compiler/aot/tfcompile.bzl

@@ -273,7 +273,6 @@ def tf_library(
             "//tensorflow/compiler/tf2xla/kernels:index_ops_kernel_argmax_float_1d",
             "//tensorflow/compiler/tf2xla/kernels:index_ops_kernel_argmax_float_2d",
             "//tensorflow/compiler/xla/service/cpu:runtime_conv2d",
-            "//tensorflow/compiler/xla/service/cpu:runtime_key_value_sort",
             "//tensorflow/compiler/xla/service/cpu:runtime_matmul",
             "//tensorflow/compiler/xla/service/cpu:runtime_single_threaded_conv2d",
             "//tensorflow/compiler/xla/service/cpu:runtime_single_threaded_matmul",