aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/labeled_tensor
diff options
context:
space:
mode:
authorGravatar Skye Wanderman-Milne <skyewm@google.com>2018-01-23 15:02:46 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-23 15:08:42 -0800
commit68a9ee1d4e041d7690a949718b2651b035a6bfad (patch)
treea93474b55a32e0ab467b01e76ee481cbf319fe8a /tensorflow/contrib/labeled_tensor
parent67ed32d35148f382de92c9383347400875eaba40 (diff)
LabeledTensor: don't test Tensor.shape object, test value instead.
With the C API enabled, a new TensorShape object may be generated every Tensor.get_shape call. PiperOrigin-RevId: 182998609
Diffstat (limited to 'tensorflow/contrib/labeled_tensor')
-rw-r--r--tensorflow/contrib/labeled_tensor/python/ops/core_test.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/labeled_tensor/python/ops/core_test.py b/tensorflow/contrib/labeled_tensor/python/ops/core_test.py
index 1f4a3ef568..e70b492374 100644
--- a/tensorflow/contrib/labeled_tensor/python/ops/core_test.py
+++ b/tensorflow/contrib/labeled_tensor/python/ops/core_test.py
@@ -225,7 +225,7 @@ class LabeledTensorTest(test_util.Base):
tensor = array_ops.placeholder(dtypes.string, [None])
actual = core.LabeledTensor(tensor, ['x'])
self.assertIsNone(actual.axes['x'].size)
- self.assertIs(actual.axes['x'].value, tensor.get_shape()[0])
+ self.assertIsNone(actual.axes['x'].value.value)
def test_eq(self):
self.assertEqual(self.lt, self.lt)