dromescu
diff --git a/‎tensorflow/python/framework/function.py‎
Lines changed: 3 additions & 5 deletions b/‎tensorflow/python/framework/function.py‎
Lines changed: 3 additions & 5 deletions
diff --git a/‎tensorflow/python/framework/function_test.py‎
Lines changed: 1 addition & 1 deletion b/‎tensorflow/python/framework/function_test.py‎
Lines changed: 1 addition & 1 deletion
@@ -654,12 +654,10 @@ def update_strs(slist):
         for s in slist:
           update_str(s)
 
-      # TODO(josh11b): Switch .node to .node_def
-      for n in sorted(self._definition.node, key=lambda n: n.ret[0]):
-        update_strs(n.ret)
+      for n in sorted(self._definition.node_def, key=lambda n: n.name):
+        update_str(n.name)
         update_str(n.op)
-        update_strs(n.arg)
-        update_strs(n.dep)
+        update_strs(n.input)
         update_num(len(n.attr))
         # NOTE: protobuf map serialization does not guarantee ordering.
         for k in sorted(n.attr):
 
@@ -566,7 +566,7 @@ def testStableName(self):
     def Foo(x, y, z):
       return tf.tanh(tf.matmul(x, y) + z)
 
-    self.assertEqual("Foo_158cce4d", Foo.instantiate([tf.float32] * 3).name)
+    self.assertEqual("Foo_e0cb6030", Foo.instantiate([tf.float32] * 3).name)
 
 
 class FunctionOverloadTest(tf.test.TestCase):