Searched refs:partial_pivoting (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tridiagonal_ops.cc | 33 bool partial_pivoting = false; in Compile() local
35 GetNodeAttr(def(), "partial_pivoting", &partial_pivoting)); in Compile()
36 if (partial_pivoting) { in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TridiagonalSolve.pbtxt | 21 name: "partial_pivoting"
40 pivoting, depending on `partial_pivoting` attribute. On GPU, Nvidia's cuSPARSE
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | tridiagonal_solve_ops_test.py | 67 linalg_impl.tridiagonal_solve(diags, rhs, partial_pivoting=False),
111 linalg_impl.tridiagonal_solve(diags, rhs, partial_pivoting=False),
156 partial_pivoting=False)),
173 partial_pivoting=False)),
236 linalg_impl.tridiagonal_solve(diags, rhs, partial_pivoting=True),
405 partial_pivoting=False)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 454 partial_pivoting=True): argument
548 conjugate_rhs, partial_pivoting,
581 conjugate_rhs, partial_pivoting,
595 diagonals, rhs, transpose_rhs, conjugate_rhs, partial_pivoting, name)
601 conjugate_rhs, partial_pivoting, name): argument
636 linalg_ops.tridiagonal_solve(diagonals, rhs, partial_pivoting, name),
645 return linalg_ops.tridiagonal_solve(diagonals, rhs, partial_pivoting, name)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TridiagonalSolve.pbtxt | 43 name: "partial_pivoting"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TridiagonalSolve.pbtxt | 43 name: "partial_pivoting"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tridiagonal_solve_op_test.py | 93 partial_pivoting=pivoting)
116 diags, rhs, diags_format, partial_pivoting=pivoting)
565 diags, rhs, diags_format, partial_pivoting=self.pivoting)
650 partial_pivoting=self.pivoting)
710 diags, rhs, partial_pivoting=pivoting)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 983 partial_pivoting = op.get_attr("partial_pivoting")
992 partial_pivoting=partial_pivoting)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 265 …agonals_format\', \'transpose_rhs\', \'conjugate_rhs\', \'name\', \'partial_pivoting\'], varargs=N…
|
| D | tensorflow.raw_ops.pbtxt | 5161 …argspec: "args=[\'diagonals\', \'rhs\', \'partial_pivoting\', \'name\'], varargs=None, keywords=No…
|
| /external/tensorflow/ |
| D | RELEASE.md | 2114 * Added partial_pivoting input parameter to `tf.linalg.tridiagonal_solve`.
2147 * Pass partial_pivoting to the `_TridiagonalSolveGrad`.
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 5161 …argspec: "args=[\'diagonals\', \'rhs\', \'partial_pivoting\', \'name\'], varargs=None, keywords=No…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 16981 pivoting, depending on `partial_pivoting` attribute. On GPU, Nvidia's cuSPARSE
16994 DefaultValuedAttr<BoolAttr, "true">:$partial_pivoting
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 56824 name: "partial_pivoting"
|