| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_ops.py | 43 def _RegularizedGramianCholesky(matrix, l2_regularizer, first_kind): argument
69 if isinstance(l2_regularizer, ops.Tensor) or l2_regularizer != 0:
80 gramian += l2_regularizer * identity
247 def matrix_solve_ls(matrix, rhs, l2_regularizer=0.0, fast=True, name=None): argument
338 def _overdetermined(matrix, rhs, l2_regularizer): argument
341 matrix, l2_regularizer=l2_regularizer, first_kind=True)
344 def _underdetermined(matrix, rhs, l2_regularizer): argument
347 matrix, l2_regularizer=l2_regularizer, first_kind=False)
350 def _composite_impl(matrix, rhs, l2_regularizer): argument
352 with ops.name_scope(name, 'matrix_solve_ls', [matrix, rhs, l2_regularizer]):
[all …]
|
| D | linalg_grad.py | 587 l2_regularizer = math_ops.cast(op.inputs[2], a.dtype.base_dtype)
590 a, l2_regularizer=l2_regularizer, first_kind=True)
611 l2_regularizer = math_ops.cast(op.inputs[2], a.dtype.base_dtype)
614 a, l2_regularizer=l2_regularizer, first_kind=False)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_solve_ls_op_test.py | 61 def _SolveWithNumpy(matrix, rhs, l2_regularizer=0): argument
62 if l2_regularizer == 0:
69 preconditioner = l2_regularizer * np.identity(cols)
74 preconditioner = l2_regularizer * np.identity(rows)
88 l2_regularizer, argument
90 if not fast and l2_regularizer != 0:
107 np_ans = _SolveWithNumpy(x, y, l2_regularizer=l2_regularizer)
120 a_ph, b_ph, fast=fast, l2_regularizer=l2_regularizer)
123 a, b, fast=fast, l2_regularizer=l2_regularizer)
134 if l2_regularizer == 0:
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_solve_ls_op_impl.h | 85 const double l2_regularizer = l2_regularizer_in.scalar<double>()(); in ComputeMatrix() local
86 OP_REQUIRES(context, l2_regularizer >= 0, in ComputeMatrix()
112 if (l2_regularizer > 0) { in ComputeMatrix()
114 (Scalar(l2_regularizer) * Matrix::Ones(cols, 1)).asDiagonal(); in ComputeMatrix()
133 if (l2_regularizer > 0) { in ComputeMatrix()
135 (Scalar(l2_regularizer) * Matrix::Ones(rows, 1)).asDiagonal(); in ComputeMatrix()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MatrixSolveLs.pbtxt | 12 name: "l2_regularizer"
48 name: "l2_regularizer"
86 name: "l2_regularizer"
|
| D | BatchMatrixSolveLs.pbtxt | 12 name: "l2_regularizer"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MatrixSolveLs.pbtxt | 12 name: "l2_regularizer"
48 name: "l2_regularizer"
86 name: "l2_regularizer"
|
| D | BatchMatrixSolveLs.pbtxt | 12 name: "l2_regularizer"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixSolveLs.pbtxt | 16 name: "l2_regularizer"
47 `l2_regularizer`=\\(\lambda \in \mathbb{R}\\).
66 `l2_regularizer` is ignored.
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 522 ShapeHandle l2_regularizer; in __anonde8650310802() local
523 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &l2_regularizer)); in __anonde8650310802()
|
| D | ops.pbtxt | 3953 name: "l2_regularizer"
23942 name: "l2_regularizer"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 165 …argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywor…
|
| D | tensorflow.pbtxt | 1669 …argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywor…
|
| D | tensorflow.raw_ops.pbtxt | 401 …argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywor…
2433 …argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywor…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.pbtxt | 177 …argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywor…
|
| D | tensorflow.raw_ops.pbtxt | 401 …argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywor…
2433 …argspec: "args=[\'matrix\', \'rhs\', \'l2_regularizer\', \'fast\', \'name\'], varargs=None, keywor…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 5192 name: "l2_regularizer"
5228 name: "l2_regularizer"
5264 name: "l2_regularizer"
16279 name: "l2_regularizer"
16315 name: "l2_regularizer"
|
| D | ops_history.v2.pbtxt | 9608 name: "l2_regularizer"
34569 name: "l2_regularizer"
34605 name: "l2_regularizer"
34643 name: "l2_regularizer"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 15360 func MatrixSolveLs(scope *Scope, matrix tf.Output, rhs tf.Output, l2_regularizer tf.Output, optiona…
15371 matrix, rhs, l2_regularizer,
|