Fix broken test builds

Change: 146316019
dontkme · Feb 2, 2017 · ffc6677 · ffc6677
1 parent fbf32f8
commit ffc6677
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/tensorflow/compiler/jit/mark_for_compilation_pass_test.cc b/tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
@@ -338,9 +338,9 @@ TEST(XlaCompilationTest, Loops) {
   auto a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT);
   auto b = ops::Placeholder(root.WithOpName("B"), DT_FLOAT);
   auto c = ops::Add(root.WithOpName("C"), a, b);
-  auto enter = ops::internal::Enter(root, c, "aframe");
+  auto enter = ops::Enter(root, c, "aframe");
   auto next_iter = ops::NextIteration(root, enter);
-  auto exit = ops::internal::Exit(root, next_iter);
+  auto exit = ops::Exit(root, next_iter);
   auto d = ops::Add(root.WithOpName("D"), c, exit);
 
   std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()));

diff --git a/tensorflow/core/graph/graph_partition_test.cc b/tensorflow/core/graph/graph_partition_test.cc
@@ -338,9 +338,9 @@ TEST_F(GraphPartitionTest, CrossDevice_DataControl) {
 TEST_F(GraphPartitionTest, CrossDeviceLoop) {
   using namespace ::tensorflow::ops;  // NOLINT(build/namespaces)
   auto a1 = BoolInput(in_.WithOpName("A1"));
-  auto a2 = ::tensorflow::ops::internal::Enter(in_.WithOpName("A2"), a1, "foo");
-  auto a3 = ::tensorflow::ops::internal::Merge(in_.WithOpName("A3"),
-                                               {a2, Input("A5", 0, DT_BOOL)})
+  auto a2 = ::tensorflow::ops::Enter(in_.WithOpName("A2"), a1, "foo");
+  auto a3 = ::tensorflow::ops::Merge(in_.WithOpName("A3"),
+                                     {a2, Input("A5", 0, DT_BOOL)})
                 .output;
   LoopCond(in_.WithOpName("A4"), a3);
   auto b1 = Identity(in_.WithOpName("B1"), a3);
@@ -352,9 +352,9 @@ TEST_F(GraphPartitionTest, CrossDeviceLoop) {
 TEST_F(GraphPartitionTest, CrossDeviceLoop1) {
   using namespace ::tensorflow::ops;  // NOLINT(build/namespaces)
   auto a1 = BoolInput(in_.WithOpName("A1"));
-  auto a2 = ::tensorflow::ops::internal::Enter(in_.WithOpName("B2"), a1, "foo");
-  auto a3 = ::tensorflow::ops::internal::Merge(in_.WithOpName("A3"),
-                                               {a2, Input("B5", 0, DT_BOOL)})
+  auto a2 = ::tensorflow::ops::Enter(in_.WithOpName("B2"), a1, "foo");
+  auto a3 = ::tensorflow::ops::Merge(in_.WithOpName("A3"),
+                                     {a2, Input("B5", 0, DT_BOOL)})
                 .output;
   LoopCond(in_.WithOpName("A4"), a3);
   auto b1 = Identity(in_.WithOpName("B1"), a3);