Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dtridiagonal_solve_op_test.py60 result = linalg_impl.tridiagonal_solve(diags, rhs, diags_format,
77 linalg_impl.tridiagonal_solve(diags, rhs, diags_format)
340 x = linalg_impl.tridiagonal_solve(diags, rhs, diags_format)
408 x = linalg_impl.tridiagonal_solve((superdiag, diag, subdiag),
444 x = linalg_impl.tridiagonal_solve(diags, rhs, transpose_rhs=True)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py336 def tridiagonal_solve(diagonals, function
509 linalg_ops.tridiagonal_solve(diagonals, rhs, name), -1)
517 result = linalg_ops.tridiagonal_solve(diagonals, rhs, name)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt208 name: "tridiagonal_solve"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt208 name: "tridiagonal_solve"