Home
last modified time | relevance | path

Searched refs:matrix_solve_with_broadcast (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dlinear_operator_util_test.py196 result = linear_operator_util.matrix_solve_with_broadcast(matrix, rhs)
213 result = linear_operator_util.matrix_solve_with_broadcast(matrix, rhs)
233 result = linear_operator_util.matrix_solve_with_broadcast(matrix_ph, rhs_ph)
250 result = linear_operator_util.matrix_solve_with_broadcast(
267 linear_operator_util.matrix_solve_with_broadcast(matrix_ph, rhs_ph),
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinear_operator_low_rank_update.py476 capinv_vh_linv_rhs = linear_operator_util.matrix_solve_with_broadcast(
H A Dlinear_operator_util.py363 def matrix_solve_with_broadcast(matrix, rhs, adjoint=False, name=None): function
H A Dlinear_operator_test_util.py412 mat_solve = linear_operator_util.matrix_solve_with_broadcast(mat_a, mat_b)
675 mat_solve = linear_operator_util.matrix_solve_with_broadcast(
H A Dlinear_operator.py798 return linear_operator_util.matrix_solve_with_broadcast(