See https://openxla.org/stablehlo/spec#iota for details.
Usage
infer_types_iota(iota_dimension, dtype, shape)
hlo_iota(iota_dimension, dtype, shape, func = NULL)Arguments
- iota_dimension
(
integer(1))
The dimension along which to generate increasing values. Must be in range[0, rank(output)).- dtype
(
character(1))
The data type of the output tensor. One of: pred, i8, i16, i32, i64, ui8, ui16, ui32, ui64, f32, f64 (excluding boolean).- shape
(
integer())
The shape of the output tensor.- func
(
Func)
The function to add the operation to. Per default, uses the last function created withhlo_funcorlocal_func.