Searched refs:DoMatrixTranspose (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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 | 174 DoMatrixTranspose(device, *packed_triangular_factors, in ComputeAsync()
218 DoMatrixTranspose(device, packed_triangular_factors_transpose, in ComputeAsync()
|
| D | svd_op_gpu.cu.cc | 258 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, u_copy, U), done); in RunSVD()
260 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, v_copy, V), in RunSVD()
306 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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | 226 Status DoMatrixTranspose(const GPUDevice& device, const Tensor& in, in DoMatrixTranspose() function
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 478 DoMatrixTranspose(ctx->eigen_device<CPUDevice>(), input, output)); in TransposeAndConjugateAllocatedTensor()
653 ctx, DoMatrixTranspose(d, b_t /*input*/, &b_t_input /*output*/)); in Compute()
681 OP_REQUIRES_OK(ctx, DoMatrixTranspose(d, /*input=*/c_mat_col_major_t, in Compute()
|