Mojo function
gather_elementwise_fn_wrapper
gather_elementwise_fn_wrapper[*, dtype: DType, indices_type: DType, input_fn: fn[width: Int, rank: Int](IndexList[rank]) capturing -> SIMD[dtype, width], indices_fn: fn[width: Int, rank: Int](IndexList[rank]) capturing -> SIMD[indices_type, width], output_fn: fn[width: Int, rank: Int](IndexList[rank], SIMD[dtype, width]) capturing -> None, simd_width: Int, prefetch_fn: OptionalReg[fn[input_rank: Int, indices_rank: Int](IndexList[input_rank], IndexList[indices_rank]) capturing -> None] = OptionalReg[fn[input_rank: Int, indices_rank: Int](IndexList[input_rank], IndexList[indices_rank]) capturing -> None]({:i1 0, 1}), error_index_fn: OptionalReg[fn(Int) capturing -> None] = OptionalReg[fn(Int) capturing -> None]({:i1 0, 1})](axis: Axis, input_shape: IndexList[size, element_type=element_type], indices_shape: IndexList[size, element_type=element_type], output_shape: IndexList[size, element_type=element_type], coords: IndexList[size, element_type=element_type])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!