| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_solve_op.cc | 50 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_)); in MatrixSolveOp()
85 if (adjoint_) { in ComputeMatrix()
112 bool adjoint_; member in tensorflow::MatrixSolveOp
125 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_)); in MatrixSolveOpGpu()
177 if (adjoint_) { in ComputeAsync()
298 solver->GetrsBatched(adjoint_ ? CUBLAS_OP_C : CUBLAS_OP_T, n, nrhs, in ComputeAsync()
314 solver->Getrs(adjoint_ ? CUBLAS_OP_C : CUBLAS_OP_T, n, nrhs, in ComputeAsync()
357 bool adjoint_; member in tensorflow::MatrixSolveOpGpu
|
| D | matrix_inverse_op.cc | 48 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_)); in MatrixInverseOp()
59 if (adjoint_) { in ComputeMatrix()
81 bool adjoint_; member in tensorflow::MatrixInverseOp
95 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_)); in MatrixInverseOpGpu()
140 if (!adjoint_) { in ComputeAsync()
266 bool adjoint_; member in tensorflow::MatrixInverseOpGpu
|
| D | banded_triangular_solve_op.cc | 210 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_)); in BandedTriangularSolveOpCpu()
250 " ", lower_, " ", adjoint_)); in Compute()
264 ctx, in0_reshaped, in1_reshaped, adjoint_, lower_, bcast, in Compute()
280 bool adjoint_; member in tensorflow::BandedTriangularSolveOpCpu
|
| D | matrix_triangular_solve_op_impl.h | 155 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_));
191 if (adjoint_) std::swap(d0, d1);
196 " ", lower_, " ", adjoint_));
210 ctx, in0_reshaped, in1_reshaped, adjoint_, lower_, bcast,
218 bool adjoint_;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_inverse_op.cc | 27 OP_REQUIRES_OK(ctx, ctx->GetAttr("adjoint", &adjoint_)); in MatrixInverseOp()
42 xla::XlaOp input = xla::MaybeTransposeInMinorDims(ctx->Input(0), adjoint_); in Compile()
58 bool adjoint_; member in tensorflow::__anon8735d2ab0111::MatrixInverseOp
|
| D | matrix_solve_op.cc | 28 OP_REQUIRES_OK(ctx, ctx->GetAttr("adjoint", &adjoint_)); in MatrixSolveOp()
60 xla::BatchDot(inv, adjoint_, rhs, in Compile()
66 bool adjoint_; member in tensorflow::__anon3819edf80111::MatrixSolveOp
|
| D | matrix_triangular_solve_op.cc | 32 OP_REQUIRES_OK(ctx, ctx->GetAttr("adjoint", &adjoint_)); in MatrixTriangularSolveOp()
68 adjoint_ ? xla::TriangularSolveOptions::ADJOINT in Compile()
78 bool adjoint_; member in tensorflow::__anone9e59ef00111::MatrixTriangularSolveOp
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_inverse_op_test.py | 144 for adjoint_ in True, False:
147 inv1 = linalg_ops.matrix_inverse(matrix1, adjoint=adjoint_)
148 inv2 = linalg_ops.matrix_inverse(matrix2, adjoint=adjoint_)
|
| D | matrix_solve_op_test.py | 128 for adjoint_ in False, True:
137 s1 = linalg_ops.matrix_solve(lhs1, rhs1, adjoint=adjoint_)
138 s2 = linalg_ops.matrix_solve(lhs2, rhs2, adjoint=adjoint_)
|