Home
last modified time | relevance | path

Searched refs:matrix_triangular_solve (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dmatrix_triangular_solve_op_test.py62 x = linalg_ops.matrix_triangular_solve(
149 linalg_ops.matrix_triangular_solve(a, b)
159 linalg_ops.matrix_triangular_solve(lhs, rhs)
161 linalg_ops.matrix_triangular_solve(lhs, rhs)
171 linalg_ops.matrix_triangular_solve(lhs, rhs)
173 linalg_ops.matrix_triangular_solve(lhs, rhs)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dlinalg_ops.py84 def matrix_triangular_solve(matrix, rhs, lower=True, adjoint=False, name=None): function
139 return gen_linalg_ops.matrix_triangular_solve(
185 y = gen_linalg_ops.matrix_triangular_solve(
187 x = gen_linalg_ops.matrix_triangular_solve(
H A Dlinalg_grad.py469 l_inverse = linalg_ops.matrix_triangular_solve(l,
510 linalg_ops.matrix_triangular_solve(
689 grad_b = linalg_ops.matrix_triangular_solve(
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dcholesky_op_test.py59 l_inverse_middle = linalg_ops.matrix_triangular_solve(l, middle, adjoint=True)
65 grad_a = linalg_ops.matrix_triangular_solve(l, z_h, adjoint=True)
78 l_inverse = linalg_ops.matrix_triangular_solve(l,
H A Dmatrix_triangular_solve_op_test.py85 tf_ans = linalg_ops.matrix_triangular_solve(
231 ans = linalg_ops.matrix_triangular_solve(singular_matrix, rhs)
H A Dlinalg_grad_test.py204 linalg_ops.matrix_triangular_solve,
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dmath_test.py750 return linalg_ops.matrix_triangular_solve(
H A Dpfor.py3487 output = linalg_ops.matrix_triangular_solve(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinalg_impl.py62 triangular_solve = linalg_ops.matrix_triangular_solve
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.pbtxt1688 name: "matrix_triangular_solve"