Mojo function

gather_elementwise_fn_wrapper

gather_elementwise_fn_wrapper[dtype: DType, indices_type: DType, //, *, input_fn: def[width: Int, rank: Int](IndexList[rank]) capturing -> SIMD[dtype, width], indices_fn: def[width: Int, rank: Int](IndexList[rank]) capturing -> SIMD[indices_type, width], output_fn: def[width: Int, rank: Int](IndexList[rank], SIMD[dtype, width]) capturing -> None, simd_width: Int, prefetch_fn: OptionalReg[def[input_rank: Int, indices_rank: Int](IndexList[input_rank], IndexList[indices_rank]) capturing -> None] = None, error_index_fn: Optional[error_index_fn_type] = None](axis: Axis, input_shape: IndexList[input_shape.size, element_type=input_shape.element_type], indices_shape: IndexList[indices_shape.size, element_type=indices_shape.element_type], output_shape: IndexList[output_shape.size, element_type=output_shape.element_type], coords: IndexList[coords.size, element_type=coords.element_type])

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit