diff options
author | 2017-08-15 10:12:20 -0700 | |
---|---|---|
committer | 2017-08-15 10:16:12 -0700 | |
commit | 22eab4bb57fad99a338bd2ec1ad61e5044a50f75 (patch) | |
tree | 2a37c98b781ddde43b199b6c68fa6c7ba914a34c /tensorflow/python/saved_model | |
parent | 88bde521d88b61e3bd7e298cdd79b1e1ab2229fe (diff) |
Simplify saved_model/utils_test by using tf.sparse_placeholder() as a quick
way of getting a SparseTensor of a particular shape.
PiperOrigin-RevId: 165322708
Diffstat (limited to 'tensorflow/python/saved_model')
-rw-r--r-- | tensorflow/python/saved_model/utils_test.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/python/saved_model/utils_test.py b/tensorflow/python/saved_model/utils_test.py index 85e6757b26..c9b38ed603 100644 --- a/tensorflow/python/saved_model/utils_test.py +++ b/tensorflow/python/saved_model/utils_test.py @@ -38,9 +38,7 @@ class UtilsTest(test.TestCase): self.assertEqual(1, x_tensor_info.tensor_shape.dim[0].size) def testBuildTensorInfoSparse(self): - x = sparse_tensor.SparseTensor(indices=[[3, 3], [4, 4], [5, 5]], - values=[103.0, 104.0, 105.0], - dense_shape=[42, 69]) + x = array_ops.sparse_placeholder(dtypes.float32, [42, 69], name="x") x_tensor_info = utils.build_tensor_info(x) self.assertEqual(x.values.name, x_tensor_info.coo_sparse.values_tensor_name) |