Reduce Operator — infer_types_reduce • stablehlo

See https://openxla.org/stablehlo/spec#reduce for details.

Usage

infer_types_reduce(..., body, dimensions)

hlo_reduce(inputs, init_values, dimensions, body)

Arguments

...: (Inputs, Init values)
inputs, init_values, dimensions, body: (FuncValue)

Value