gather_axis.h File Reference

Go to the source code of this file.

Functions
template<typename T, typename IdxT, typename LocT, bool SrcC, bool IdxC>
void	gather_axis (const device T *src, const device IdxT *indices, device T *out, const constant int *shape, const constant int64_t *src_strides, const constant int64_t *idx_strides, const constant size_t &ndim, const constant int &axis, const constant int &axis_size, const constant size_t &src_ax_stride, const constant size_t &idx_ax_stride, uint3 index, uint3 grid_dim)

Function Documentation

gather_axis()

template<typename T, typename IdxT, typename LocT, bool SrcC, bool IdxC>

void gather_axis	(	const device T *	src,
		const device IdxT *	indices,
		device T *	out,
		const constant int *	shape,
		const constant int64_t *	src_strides,
		const constant int64_t *	idx_strides,
		const constant size_t &	ndim,
		const constant int &	axis,
		const constant int &	axis_size,
		const constant size_t &	src_ax_stride,
		const constant size_t &	idx_ax_stride,
		uint3	index,
		uint3	grid_dim )