提交 eeb2632c 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Switch function definition hash to NodeDef, fixing a TODO.

Change: 139963371
上级 6e45225c
......@@ -654,12 +654,10 @@ class _DefinedFunction(object):
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 @@ class FunctionTest(tf.test.TestCase):
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):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册