Searched refs:DoMatrixTranspose (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | qr_op_impl.h | 199 context, DoMatrixTranspose(device, input, &input_transposed), done); in ComputeAsync()
222 context, DoMatrixTranspose(device, input_transposed, r), done); in ComputeAsync()
284 context, DoMatrixTranspose(device, input_transposed, q), done); in ComputeAsync()
|
| D | matrix_solve_op.cc | 186 DoMatrixTranspose(device, input, &input_copy), done); in ComputeAsync()
263 context, DoMatrixTranspose(device, rhs, &transposed_rhs), done); in ComputeAsync()
326 context, DoMatrixTranspose(device, transposed_rhs, output), done); in ComputeAsync()
|
| D | lu_op_gpu.cu.cc | 172 DoMatrixTranspose(device, *packed_triangular_factors, in ComputeAsync()
216 DoMatrixTranspose(device, packed_triangular_factors_transpose, in ComputeAsync()
|
| D | svd_op_gpu.cu.cc | 211 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, u_copy, U), done); in RunSVD()
255 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, M, &input_copy), in PerformSVD_MgeqN()
|
| D | self_adjoint_eig_v2_op_gpu.cc | 152 context, DoMatrixTranspose(device, input_copy, eigenvectors), done); in ComputeAsync()
|
| D | transpose_functor_cpu.cc | 125 Status DoMatrixTranspose(const DEVICE& device, const Tensor& in, \
|
| D | transpose_functor.h | 52 Status DoMatrixTranspose(const Device& device, const Tensor& in, Tensor* out);
|
| D | transpose_functor_gpu.cu.cc | 225 Status DoMatrixTranspose(const GPUDevice& device, const Tensor& in, in DoMatrixTranspose() function
|