From 4cedc8b6e738b7a188c9c091cf667bacafae44b7 Mon Sep 17 00:00:00 2001 From: Alexandre Passos Date: Thu, 27 Sep 2018 13:18:33 -0700 Subject: Updating the V2 variables API. PiperOrigin-RevId: 214824023 --- tensorflow/python/kernel_tests/identity_op_py_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/python/kernel_tests/identity_op_py_test.py') diff --git a/tensorflow/python/kernel_tests/identity_op_py_test.py b/tensorflow/python/kernel_tests/identity_op_py_test.py index 37f9f716f8..88ea10c22a 100644 --- a/tensorflow/python/kernel_tests/identity_op_py_test.py +++ b/tensorflow/python/kernel_tests/identity_op_py_test.py @@ -61,7 +61,7 @@ class IdentityOpTest(test.TestCase): def testRefIdentityShape(self): with self.cached_session(): shape = [2, 3] - tensor = variables.Variable( + tensor = variables.VariableV1( constant_op.constant( [[1, 2, 3], [6, 5, 4]], dtype=dtypes.int32)) self.assertEquals(shape, tensor.get_shape()) -- cgit v1.2.3