unary.h File Reference

Go to the source code of this file.

Functions
template<typename T, typename U, typename Op>
void	unary_v (device const T *in, device U *out, uint index)
template<typename T, typename U, typename Op>
void	unary_v2 (device const T *in, device U *out, uint2 index, uint2 grid_dim)
template<typename T, typename U, typename Op, int N = 1, typename IdxT = int64_t>
void	unary_g (device const T *in, device U *out, constant const int *in_shape, constant const int64_t *in_strides, device const int &ndim, uint3 index, uint3 grid_dim)

Function Documentation

unary_g()

template<typename T, typename U, typename Op, int N = 1, typename IdxT = int64_t>

void unary_g	(	device const T *	in,
		device U *	out,
		constant const int *	in_shape,
		constant const int64_t *	in_strides,
		device const int &	ndim,
		uint3	index,
		uint3	grid_dim )

unary_v()

template<typename T, typename U, typename Op>

void unary_v	(	device const T *	in,
		device U *	out,
		uint	index )

unary_v2()

template<typename T, typename U, typename Op>

void unary_v2	(	device const T *	in,
		device U *	out,
		uint2	index,
		uint2	grid_dim )