Part 2c of renaming SparseTensor.shape -> SparseTensor.dense_shape

Change: 141066682

1 files changed, 7 insertions, 4 deletions

diff --git a/tensorflow/python/kernel_tests/parsing_ops_test.py b/tensorflow/python/kernel_tests/parsing_ops_test.py
index badd10c4cb..56bef40301 100644
--- a/tensorflow/python/kernel_tests/parsing_ops_test.py
+++ b/tensorflow/python/kernel_tests/parsing_ops_test.py
@@ -47,7 +47,8 @@ def flatten(list_of_lists):
 def flatten_values_tensors_or_sparse(tensors_list):
   """Flatten each SparseTensor object into 3 Tensors for session.run()."""
   return list(
-      flatten([[v.indices, v.values, v.shape] if isinstance(v, tf.SparseTensor)
+      flatten([[v.indices, v.values, v.dense_shape]
+               if isinstance(v, tf.SparseTensor)
                else [v] for v in tensors_list]))
 
 
@@ -102,7 +103,8 @@ class ParseExampleTest(tf.test.TestCase):
           self.assertEqual(
               tuple(out[k].indices.get_shape().as_list()), (None, 2))
           self.assertEqual(tuple(out[k].values.get_shape().as_list()), (None,))
-          self.assertEqual(tuple(out[k].shape.get_shape().as_list()), (2,))
+          self.assertEqual(
+              tuple(out[k].dense_shape.get_shape().as_list()), (2,))
 
   def testEmptySerializedWithAllDefaults(self):
     sparse_name = "st_a"
@@ -601,7 +603,8 @@ class ParseSingleExampleTest(tf.test.TestCase):
           self.assertEqual(
               tuple(out[k].indices.get_shape().as_list()), (None, 1))
           self.assertEqual(tuple(out[k].values.get_shape().as_list()), (None,))
-          self.assertEqual(tuple(out[k].shape.get_shape().as_list()), (1,))
+          self.assertEqual(
+              tuple(out[k].dense_shape.get_shape().as_list()), (1,))
 
   def testSingleExampleWithSparseAndSparseFeatureAndDense(self):
     original = example(features=features({"c": float_feature([3, 4]),
@@ -716,7 +719,7 @@ class ParseSequenceExampleTest(tf.test.TestCase):
             self.assertEqual(
                 tuple(context_out[k].values.get_shape().as_list()), (None,))
             self.assertEqual(
-                tuple(context_out[k].shape.get_shape().as_list()), (1,))
+                tuple(context_out[k].dense_shape.get_shape().as_list()), (1,))
 
   def testSequenceExampleWithSparseAndDenseContext(self):
     original = sequence_example(context=features({"c": float_feature([3, 4]),