Home
last modified time | relevance | path

Searched refs:tridiagonal_solve (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dtridiagonal_solve_ops_test.py67 linalg_impl.tridiagonal_solve(diags, rhs, partial_pivoting=False),
111 linalg_impl.tridiagonal_solve(diags, rhs, partial_pivoting=False),
150 linalg_impl.tridiagonal_solve(
168 linalg_impl.tridiagonal_solve([_tfconst(x) for x in diags],
236 linalg_impl.tridiagonal_solve(diags, rhs, partial_pivoting=True),
399 x = linalg_impl.tridiagonal_solve(
472 linalg_impl.tridiagonal_solve(diags, rhs, diags_format)
/external/tensorflow/tensorflow/python/kernel_tests/
Dtridiagonal_solve_op_test.py87 result = linalg_impl.tridiagonal_solve(
115 linalg_impl.tridiagonal_solve(
408 x = linalg_impl.tridiagonal_solve(
564 x = linalg_impl.tridiagonal_solve(
647 x = linalg_impl.tridiagonal_solve((superdiag, diag, subdiag),
709 x = linalg_impl.tridiagonal_solve(
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py448 def tridiagonal_solve(diagonals, function
636 linalg_ops.tridiagonal_solve(diagonals, rhs, partial_pivoting, name),
645 return linalg_ops.tridiagonal_solve(diagonals, rhs, partial_pivoting, name)
Dlinear_operator_tridiag.py336 y = linalg.tridiagonal_solve(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt264 name: "tridiagonal_solve"
/external/tensorflow/tensorflow/python/ops/
Dlinalg_grad.py991 grad_rhs = linalg_ops.tridiagonal_solve(diags_transposed, grad,
/external/tensorflow/
DRELEASE.md2113 * Added `tf.linalg.tridiagonal_solve` op.
2114 * Added partial_pivoting input parameter to `tf.linalg.tridiagonal_solve`.
2115 * Added gradient to `tf.linalg.tridiagonal_solve`.
2401 * Add tf.linalg.tridiagonal_solve op.
2415 * Added GPU implementation of tf.linalg.tridiagonal_solve.