diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-10-17 16:19:08 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-17 16:25:58 -0700 |
commit | 47e4d4b6b5742350233a8fd83cd81269792ed286 (patch) | |
tree | e13ba390de56684359e9771a98dd80690dfd1121 /tensorflow/core/kernels/self_adjoint_eig_v2_op_gpu.cc | |
parent | 95c7f5344f8da74a839c459c6415855bffe4f004 (diff) |
Use optimized functor for conjugate transpose in MatrixInverseOp.
Introduce convenience functions DoMatrixTranspose and DoConjugateMatrixTranspose.
Misc. minor cleanup of templates in transpose_functor*.
PiperOrigin-RevId: 172532252
Diffstat (limited to 'tensorflow/core/kernels/self_adjoint_eig_v2_op_gpu.cc')
-rw-r--r-- | tensorflow/core/kernels/self_adjoint_eig_v2_op_gpu.cc | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/self_adjoint_eig_v2_op_gpu.cc b/tensorflow/core/kernels/self_adjoint_eig_v2_op_gpu.cc index b0b4f89a27..3a84df07a9 100644 --- a/tensorflow/core/kernels/self_adjoint_eig_v2_op_gpu.cc +++ b/tensorflow/core/kernels/self_adjoint_eig_v2_op_gpu.cc @@ -148,11 +148,8 @@ class SelfAdjointEigV2OpGpu : public AsyncOpKernel { if (compute_v_) { // Transpose eigenvectors now stored in input_copy in column-major form to // output in row-major form. - std::vector<int> perm(ndims); - std::iota(perm.begin(), perm.end(), 0); - std::swap(perm[ndims - 2], perm[ndims - 1]); OP_REQUIRES_OK_ASYNC( - context, DoTranspose(device, input_copy, perm, eigenvectors), done); + context, DoMatrixTranspose(device, input_copy, eigenvectors), done); } // Asynchronously check return status from cuSolver kernels. |