ternary.h File Reference

Go to the source code of this file.

Functions
template<typename T, typename Op>
void	ternary_v (device const bool *a, device const T *b, device const T *c, device T *d, uint index)
template<typename T, typename Op>
void	ternary_v2 (device const bool *a, device const T *b, device const T *c, device T *d, uint2 index, uint2 grid_dim)
template<typename T, typename Op, typename IdxT = int64_t>
void	ternary_g_nd1 (device const bool *a, device const T *b, device const T *c, device T *d, constant const int64_t &a_strides, constant const int64_t &b_strides, constant const int64_t &c_strides, uint index)
template<typename T, typename Op, typename IdxT = int64_t>
void	ternary_g_nd2 (device const bool *a, device const T *b, device const T *c, device T *d, constant const int64_t a_strides[2], constant const int64_t b_strides[2], constant const int64_t c_strides[2], uint2 index, uint2 grid_dim)
template<typename T, typename Op, typename IdxT = int64_t>
void	ternary_g_nd3 (device const bool *a, device const T *b, device const T *c, device T *d, constant const int64_t a_strides[3], constant const int64_t b_strides[3], constant const int64_t c_strides[3], uint3 index, uint3 grid_dim)
template<typename T, typename Op, int N = 1, typename IdxT = int64_t>
void	ternary_g (device const bool *a, device const T *b, device const T *c, device T *d, constant const int *shape, constant const int64_t *a_strides, constant const int64_t *b_strides, constant const int64_t *c_strides, constant const int &ndim, uint3 index, uint3 grid_dim)

Function Documentation

ternary_g()

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

void ternary_g	(	device const bool *	a,
		device const T *	b,
		device const T *	c,
		device T *	d,
		constant const int *	shape,
		constant const int64_t *	a_strides,
		constant const int64_t *	b_strides,
		constant const int64_t *	c_strides,
		constant const int &	ndim,
		uint3	index,
		uint3	grid_dim )

ternary_g_nd1()

template<typename T, typename Op, typename IdxT = int64_t>

void ternary_g_nd1	(	device const bool *	a,
		device const T *	b,
		device const T *	c,
		device T *	d,
		constant const int64_t &	a_strides,
		constant const int64_t &	b_strides,
		constant const int64_t &	c_strides,
		uint	index )

ternary_g_nd2()

template<typename T, typename Op, typename IdxT = int64_t>

void ternary_g_nd2	(	device const bool *	a,
		device const T *	b,
		device const T *	c,
		device T *	d,
		constant const int64_t	a_strides[2],
		constant const int64_t	b_strides[2],
		constant const int64_t	c_strides[2],
		uint2	index,
		uint2	grid_dim )

ternary_g_nd3()

template<typename T, typename Op, typename IdxT = int64_t>

void ternary_g_nd3	(	device const bool *	a,
		device const T *	b,
		device const T *	c,
		device T *	d,
		constant const int64_t	a_strides[3],
		constant const int64_t	b_strides[3],
		constant const int64_t	c_strides[3],
		uint3	index,
		uint3	grid_dim )

ternary_v()

template<typename T, typename Op>

void ternary_v	(	device const bool *	a,
		device const T *	b,
		device const T *	c,
		device T *	d,
		uint	index )

ternary_v2()

template<typename T, typename Op>

void ternary_v2	(	device const bool *	a,
		device const T *	b,
		device const T *	c,
		device T *	d,
		uint2	index,
		uint2	grid_dim )