| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_triangular_solve_op.cc | 51 : Base(context), lower_(true), adjoint_(false) { in MatrixTriangularSolveOp()
53 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_)); in MatrixTriangularSolveOp()
96 if (adjoint_) { in ComputeMatrix()
103 if (adjoint_) { in ComputeMatrix()
113 bool adjoint_; member in tensorflow::MatrixTriangularSolveOp
143 : Base(context), lower_(true), adjoint_(false) { in MatrixTriangularSolveOpGPU()
145 OP_REQUIRES_OK(context, context->GetAttr("adjoint", &adjoint_)); in MatrixTriangularSolveOpGPU()
212 if (adjoint_) { in ComputeMatrix()
238 bool adjoint_; member in tensorflow::MatrixTriangularSolveOpGPU
|
| 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
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_triangular_solve_op.cc | 29 OP_REQUIRES_OK(ctx, ctx->GetAttr("adjoint", &adjoint_)); in MatrixTriangularSolveOp()
37 adjoint_ ? xla::TriangularSolveOptions::ADJOINT in Compile()
44 bool adjoint_; member in tensorflow::__anondb49a8890111::MatrixTriangularSolveOp
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_inverse_op_test.py | 147 for adjoint_ in True, False:
150 inv1 = linalg_ops.matrix_inverse(matrix1, adjoint=adjoint_)
151 inv2 = linalg_ops.matrix_inverse(matrix2, adjoint=adjoint_)
|
| D | matrix_solve_op_test.py | 127 for adjoint_ in False, True:
132 s1 = linalg_ops.matrix_solve(lhs1, rhs1, adjoint=adjoint_)
133 s2 = linalg_ops.matrix_solve(lhs2, rhs2, adjoint=adjoint_)
|