Device pointer interface: Shift is host pointer |
Shift the contents of an array along specified dimensions and wrap.- Parameters:
-
[out] | d_out | Shifted array |
[in] | ndims | Number of dimensions in array dims |
[in] | dims | Array of dimensions |
[in] | nshifts | Number of shifts in array shifts |
[in] | h_shifts | Array(on host) indicating extent of shift along each dimension, e.g. [1 0 -2] shifts forward once in first dimension, none in second dimension, and back twice in third dimension. |
[in] | d_in | Original array. |
[in] | batch | Number of tiles of input |
|
afError | af_shift_B (bool *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *h_shifts, const bool *d_in, unsigned batch) |
| Boolean (8-bit) array shift.
|
afError | af_shift_I (int *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *h_shifts, const int *d_in, unsigned batch) |
| 32-bit signed integer array shift.
|
afError | af_shift_U (unsigned *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *h_shifts, const unsigned *d_in, unsigned batch) |
| 32-bit unsigned integer array shift.
|
afError | af_shift_S (float *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *h_shifts, const float *d_in, unsigned batch) |
| Single-precision array shift.
|
afError | af_shift_D (double *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *h_shifts, const double *d_in, unsigned batch) |
| Double-precision array shift.
|
afError | af_shift_C (cuComplex *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *h_shifts, const cuComplex *d_in, unsigned batch) |
| Complex single-precision array shift.
|
afError | af_shift_Z (cuDoubleComplex *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *h_shifts, const cuDoubleComplex *d_in, unsigned batch) |
| Complex double-precision array shift.
|
Device pointer interface: Shift is device pointer |
Shift the contents of an array along specified dimensions and wrap.- Parameters:
-
[out] | d_out | Shifted array |
[in] | ndims | Number of dimensions in array dims |
[in] | dims | Array of dimensions |
[in] | nshifts | Number of shifts in array shifts |
[in] | d_shifts | Array(on device) indicating extent of shift along each dimension, e.g. [1 0 -2] shifts forward once in first dimension, none in second dimension, and back twice in third dimension. |
[in] | d_in | Original array. |
[in] | batch | Number of tiles of input. |
[in] | batch_shift | Number of tiles of shift. |
|
afError | af_shiftd_B (bool *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *d_shifts, const bool *d_in, unsigned batch, unsigned batch_shift) |
| Boolean (8-bit) array shift.
|
afError | af_shiftd_I (int *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *d_shifts, const int *d_in, unsigned batch, unsigned batch_shift) |
| 32-bit signed integer array shift.
|
afError | af_shiftd_U (unsigned *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *d_shifts, const unsigned *d_in, unsigned batch, unsigned batch_shift) |
| 32-bit unsigned integer array shift.
|
afError | af_shiftd_S (float *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *d_shifts, const float *d_in, unsigned batch, unsigned batch_shift) |
| Single-precision array shift.
|
afError | af_shiftd_D (double *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *d_shifts, const double *d_in, unsigned batch, unsigned batch_shift) |
| Double-precision array shift.
|
afError | af_shiftd_C (cuComplex *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *d_shifts, const cuComplex *d_in, unsigned batch, unsigned batch_shift) |
| Complex single-precision array shift.
|
afError | af_shiftd_Z (cuDoubleComplex *d_out, unsigned ndims, const unsigned *dims, int nshifts, const int *d_shifts, const cuDoubleComplex *d_in, unsigned batch, unsigned batch_shift) |
| Complex double-precision array shift.
|