TriangularSolve Operator — infer_types_triangular

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,
  output_types = NULL
)

Arguments

a, b: (FuncValue)
left_side: (logical(1))
If TRUE, solve op(a) * x = b. If FALSE, solve x * op(a) = b.
lower: (logical(1))
If TRUE, use lower triangle of a. If FALSE, use upper triangle.
unit_diagonal: (logical(1))
If TRUE, assume diagonal elements of a are 1.
transpose_a: (character(1))
One of "NO_TRANSPOSE", "TRANSPOSE", or "ADJOINT".
output_types: (list() of ValueType | NULL)
Output types known ahead of time (e.g. from type inference at trace time). When provided, type inference and its input validation are skipped.

Value

FuncValue