diff options
| author |  Eugene Brevdo <ebrevdo@google.com> | 2016-11-01 15:39:55 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-01 16:48:13 -0700 |
| commit | 2939f53ea2d5dc9de018d9ad64a3ac03d03d4925 (patch) | |
| tree | 9cee20d757b657779cf2f54a323ec5b57f8229f1 /tensorflow/python/kernel_tests/shape_ops_test.py | |
| parent | 0a170e45d6374ab04073542d2a34af8c984afe2f (diff) | |
Move SparseTensor into its own framework.py, add get_shape().
Change: 137887950
Diffstat (limited to 'tensorflow/python/kernel_tests/shape_ops_test.py')
| -rw-r--r-- | tensorflow/python/kernel_tests/shape_ops_test.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/shape_ops_test.py b/tensorflow/python/kernel_tests/shape_ops_test.py index 465a02bfe6..67d6fdd96a 100644 --- a/tensorflow/python/kernel_tests/shape_ops_test.py +++ b/tensorflow/python/kernel_tests/shape_ops_test.py @@ -22,7 +22,6 @@ import numpy as np import tensorflow as tf -from tensorflow.python.framework import ops from tensorflow.python.ops import array_ops @@ -36,7 +35,7 @@ def _sparsify(x, thresh=0.5, index_dtype=np.int64): x_values = x[non_zero] x_shape = x.shape - return ops.SparseTensor( + return tf.SparseTensor( indices=x_indices, values=x_values, shape=x_shape), len(x_values) class ShapeOpsTest(tf.test.TestCase): |