diff options
| author |  Yangzihao Wang <yangzihao@google.com> | 2017-07-26 15:27:52 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-26 15:32:09 -0700 |
| commit | 1bad826d6f94eb2f9b619badf3e9c2d42e370e9c (patch) | |
| tree | 1dc9a3a7d0f55b9cb07267cf7fbe9c3417e09f3b /tensorflow/python/kernel_tests/transpose_op_test.py | |
| parent | f1537588d4983dfdabfdd82442264a9d1a702d2f (diff) | |
Rollback of GPU kernel implementation of transpose for tensors with one small dimension.
END_PUBLIC
BEGIN_PUBLIC
Automated g4 rollback of changelist 162525519
PiperOrigin-RevId: 163265445
Diffstat (limited to 'tensorflow/python/kernel_tests/transpose_op_test.py')
| -rw-r--r-- | tensorflow/python/kernel_tests/transpose_op_test.py | 20 |
1 files changed, 0 insertions, 20 deletions
diff --git a/tensorflow/python/kernel_tests/transpose_op_test.py b/tensorflow/python/kernel_tests/transpose_op_test.py index 67bdb4237d..570fa79944 100644 --- a/tensorflow/python/kernel_tests/transpose_op_test.py +++ b/tensorflow/python/kernel_tests/transpose_op_test.py @@ -229,26 +229,6 @@ class TransposeTest(test.TestCase): self.assertAllEqual(np_ans, tf_ans) self.assertShapeEqual(np_ans, y) - def testLargeSizeGPU(self): - # If no GPU available, skip the test - if not test.is_gpu_available(cuda_only=True): - return - - large_shapes = [[1000000, 31, 3], [3, 1000000, 31], [3, 31, 1000000], - [2, 1000, 1000], [1000, 2, 1000], [1000, 1000, 2]] - perms = [[0, 2, 1]] * 6 - - for input_shape, perm in zip(large_shapes, perms): - total_size = np.prod(input_shape) - inp = np.arange(1, total_size + 1, dtype=np.float32).reshape(input_shape) - np_ans = self._np_transpose(inp, perm) - with self.test_session(use_gpu=True): - inx = ops.convert_to_tensor(inp) - y = array_ops.transpose(inx, perm) - tf_ans = y.eval() - self.assertAllEqual(np_ans, tf_ans) - self.assertShapeEqual(np_ans, y) - def testNop(self): self._compareCpu(np.arange(0, 6).reshape([3, 2]).astype(np.float32), [0, 1]) |