BroadcastInDim Operator — infer_types_broadcast_in_dim • stablehlo

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

Usage

infer_types_broadcast_in_dim(operand, broadcast_dimensions, shape)

hlo_broadcast_in_dim(operand, broadcast_dimensions, shape, output_types = NULL)

Arguments

operand, broadcast_dimensions, shape: (FuncValue)
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