See https://openxla.org/stablehlo/spec#triangular_solve for details.
Usage
infer_types_triangular_solve(
a,
b,
left_side,
lower,
unit_diagonal,
transpose_a
)
hlo_triangular_solve(a, b, left_side, lower, unit_diagonal, transpose_a)Arguments
- a, b
- left_side
(
logical(1))
IfTRUE, solveop(a) * x = b. IfFALSE, solvex * op(a) = b.- lower
(
logical(1))
IfTRUE, use lower triangle ofa. IfFALSE, use upper triangle.- unit_diagonal
(
logical(1))
IfTRUE, assume diagonal elements ofaare 1.- transpose_a
(
character(1))
One of"NO_TRANSPOSE","TRANSPOSE", or"ADJOINT".