experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use id() for Tensor identity comparison.

Browse Source 
PiperOrigin-RevId: 263627848
This commit is contained in:
Martin Wicke 2019-08-15 13:19:13 -07:00 committed by TensorFlower Gardener
parent 5f1f68499c
commit 053f39e766
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/training/tracking/util_test.py
View File

@ -1413,8 +1413,9 @@ class TemplateTests(parameterized.TestCase, test.TestCase):
v1_save, _, v2_save, manual_scope, manual_scope_v = save_template()
six.assertCountEqual(
self,
[v1_save, v2_save, manual_scope, manual_scope_v, save_template],
trackable_utils.list_objects(save_template))
[id(v1_save), id(v2_save), id(manual_scope),
id(manual_scope_v), id(save_template)],
map(id, trackable_utils.list_objects(save_template)))
manual_dep, = manual_scope._checkpoint_dependencies
self.assertEqual("in_manual_scope", manual_dep.name)
self.assertIs(manual_scope_v, manual_dep.ref)