gather.h File Reference

#include "mlx/backend/metal/kernels/indexing.h"

Go to the source code of this file.

Functions
template<typename T , typename IdxT , int NIDX, int IDX_NDIM>
METAL_FUNC void	gather_impl (const device T *src, device T *out, const constant int *src_shape, const constant size_t *src_strides, const constant size_t &src_ndim, const constant int *slice_sizes, const constant int *axes, const thread Indices< IdxT, NIDX > &indices, uint2 index, uint2 grid_dim)

Function Documentation

gather_impl()

template<typename T , typename IdxT , int NIDX, int IDX_NDIM>

METAL_FUNC void gather_impl	(	const device T *	src,
		device T *	out,
		const constant int *	src_shape,
		const constant size_t *	src_strides,
		const constant size_t &	src_ndim,
		const constant int *	slice_sizes,
		const constant int *	axes,
		const thread Indices< IdxT, NIDX > &	indices,
		uint2	index,
		uint2	grid_dim )