diff options
author | 2016-08-31 12:41:12 -0800 | |
---|---|---|
committer | 2016-08-31 13:47:26 -0700 | |
commit | dfc3e6cb9b9fcda8e0eccafbe6e78c4e5d9fdb79 (patch) | |
tree | 5f99abf3d8a7f369533e6013237c11648a1e6c0b /tensorflow/python/kernel_tests/xent_op_test.py | |
parent | b37153ca0c3a4b24ae179095f7221a21bb460726 (diff) |
Delegate to C++ shape functions for linalg ops and several nn_ops.
Svd, BatchTriangularSolve, BatchMatrixSolveLs, and
SparseSoftmaxCrossEntropyWithLogitsShape are now more precise in shape
computation.
Change: 131867628
Diffstat (limited to 'tensorflow/python/kernel_tests/xent_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/xent_op_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/kernel_tests/xent_op_test.py b/tensorflow/python/kernel_tests/xent_op_test.py index 70b3bdcbb4..3c843d27c8 100644 --- a/tensorflow/python/kernel_tests/xent_op_test.py +++ b/tensorflow/python/kernel_tests/xent_op_test.py @@ -70,7 +70,7 @@ class XentTest(tf.test.TestCase): [[[1., 1., 1., 1.]], [[1., 2., 3., 4.]]]).astype(dtype) np_labels = np.array( [[[0., 0., 0., 1.]], [[0., .5, .5, 0.]]]).astype(dtype) - self.assertRaisesRegexp(ValueError, "must have rank 2", + self.assertRaisesRegexp(ValueError, "must be rank 2", gen_nn_ops._softmax_cross_entropy_with_logits, np_features, np_labels) |