Gather Operator — infer_types_gather • stablehlo

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

Usage

infer_types_gather(
  operand,
  start_indices,
  gather_dimension_numbers,
  slice_sizes,
  indices_are_sorted
)

hlo_gather(
  operand,
  start_indices,
  gather_dimension_numbers,
  slice_sizes,
  indices_are_sorted = FALSE,
  output_types = NULL
)

Arguments

operand, start_indices: (FuncValue)
gather_dimension_numbers: (GatherDimensionNumbers)
The gather dimension numbers.
slice_sizes: (integer())
The sizes of the slices to gather.
indices_are_sorted: (logical(1))
Whether indices are sorted.
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