Namespaces |
namespace | af |
Functions |
array | diff1 (const array &in, int dim=-1) |
| difference between elements along a dimension
|
array | diff2 (const array &in, int dim=-1) |
| difference between every other element along a dimension
|
void | grad (array &dx, array &dy, const array &in) |
| Gradient of a matrix (central differencing).
|
afError | af_besselh_S (float *d_out, unsigned k, unsigned n, const float *d_in, float nu) |
afError | af_besselh_C (float *d_out, unsigned k, unsigned n, const float *d_in, float nu) |
afError | af_besselh_D (double *d_out, unsigned k, unsigned n, const double *d_in, double nu) |
afError | af_besselh_Z (double *d_out, unsigned k, unsigned n, const double *d_in, double nu) |
afError | af_besseli_S (float *d_out, unsigned n, const float *d_in, float nu) |
afError | af_besseli_C (float *d_out, unsigned n, const float *d_in, float nu) |
afError | af_besseli_D (double *d_out, unsigned n, const double *d_in, double nu) |
afError | af_besseli_Z (double *d_out, unsigned n, const double *d_in, double nu) |
afError | af_besselj_S (float *d_out, unsigned n, const float *d_in, float nu) |
afError | af_besselj_C (cuComplex *d_out, unsigned n, const cuComplex *d_in, float nu) |
afError | af_besselj_D (double *d_out, unsigned n, const double *d_in, double nu) |
afError | af_besselj_Z (cuDoubleComplex *d_out, unsigned n, const cuDoubleComplex *d_in, double nu) |
afError | af_besselk_S (float *d_out, unsigned n, const float *d_in, float nu) |
afError | af_besselk_C (float *d_out, unsigned n, const float *d_in, float nu) |
afError | af_besselk_D (double *d_out, unsigned n, const double *d_in, double nu) |
afError | af_besselk_Z (double *d_out, unsigned n, const double *d_in, double nu) |
afError | af_bessely_S (float *d_out, unsigned n, const float *d_in, float nu) |
afError | af_bessely_C (float *d_out, unsigned n, const float *d_in, float nu) |
afError | af_bessely_D (double *d_out, unsigned n, const double *d_in, double nu) |
afError | af_bessely_Z (double *d_out, unsigned n, const double *d_in, double nu) |
|
- Parameters:
-
[in] | ndims_out | number of output dimensions |
[in] | dims_out | output dimensions |
[out] | d_out | The output of the function |
[in] | ndims_in | number of input dimensions |
[in] | dims_in | input dimensions |
[in] | d_in | The input to the function |
[in] | order | The order of the difference |
[in] | dim | The dimension along which the operation is performed |
|
afError | af_diff_S (float *d_out, unsigned ndims_in, const unsigned *dims_in, const float *d_in, int order, int dim) |
| Single precision, difference function.
|
afError | af_diff_D (double *d_out, unsigned ndims_in, const unsigned *dims_in, const double *d_in, int order, int dim) |
| Double precision difference function.
|
|
- Parameters:
-
[out] | d_dx | Output gradient along the X dimension |
[out] | d_dy | Output gradient along the Y dimension |
[in] | nx | Number of rows in the input |
[in] | ny | Number of columns in the input |
[in] | d_in | The input the the gradient function |
[in] | batch | The number of input tiles being handled |
|
afError | af_grad_B (bool *d_dx, bool *d_dy, unsigned nx, unsigned ny, const bool *d_in, unsigned batch) |
| Boolean gradient function.
|
afError | af_grad_U (unsigned *d_dx, unsigned *d_dy, unsigned nx, unsigned ny, const unsigned *d_in, unsigned batch) |
| Unsigne integer gradient function.
|
afError | af_grad_I (int *d_dx, int *d_dy, unsigned nx, unsigned ny, const int *d_in, unsigned batch) |
| Integer gradient function.
|
afError | af_grad_S (float *d_dx, float *d_dy, unsigned nx, unsigned ny, const float *d_in, unsigned batch) |
| Single precision, floating point gradient funciton.
|
afError | af_grad_D (double *d_dx, double *d_dy, unsigned nx, unsigned ny, const double *d_in, unsigned batch) |
| Double precision, floating point gradient funciton.
|
|
- Parameters:
-
d_C | Output Matrix |
numel | Number of elements in input matrices A,B |
offset | Stride required to access next element of vectors |
d_A | Left Input Matrix |
batch_A | Number of tiles in matrix A |
batch_B | Number of tiles in matrix B |
|
afError | af_crossProd_SS (float *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const float *d_B, unsigned batch_B) |
| Single Precision Cross Product - Real x Real.
|
afError | af_crossProd_SC (cuComplex *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B) |
| Single Precision Cross Product - Real x Complex.
|
afError | af_crossProd_CS (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const float *d_B, unsigned batch_B) |
| Single Precision Cross Product - Complex x Real.
|
afError | af_crossProd_CC (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B) |
| Single Precision Cross Product - Complex x Complex.
|
afError | af_crossProd_SD (float *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const double *d_B, unsigned batch_B) |
| Cross Product - Single-precision Real x Double-precision Real.
|
afError | af_crossProd_SZ (cuComplex *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B) |
| Cross Product - Single-precision Real x Double-precision Complex.
|
afError | af_crossProd_CD (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const double *d_B, unsigned batch_B) |
| Cross Product - Single-precision Complex x Double-precision Real.
|
afError | af_crossProd_CZ (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B) |
| Cross Product - Single-precision Complex x Double-precision Complex.
|
afError | af_crossProd_DS (float *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const float *d_B, unsigned batch_B) |
| Cross Product - Double-precision Real x Single-precision Real.
|
afError | af_crossProd_DC (cuComplex *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B) |
| Cross Product - Double-precision Real x Single-precision Complex.
|
afError | af_crossProd_ZS (cuComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const float *d_B, unsigned batch_B) |
| Cross Product - Double-precision Complex x Single-precision Real.
|
afError | af_crossProd_ZC (cuComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B) |
| Cross Product - Double-precision Complex x Single-precision Complex.
|
afError | af_crossProd_DD (double *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const double *d_B, unsigned batch_B) |
| Double Precision Cross Product - Real x Real.
|
afError | af_crossProd_DZ (cuDoubleComplex *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B) |
| Double Precision Cross Product - Real x Complex.
|
afError | af_crossProd_ZD (cuDoubleComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const double *d_B, unsigned batch_B) |
| Double Precision Cross Product - Complex x Real.
|
afError | af_crossProd_ZZ (cuDoubleComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B) |
| Double Precision Cross Product - Complex x Complex.
|
|
- Parameters:
-
d_C | Output Matrix |
mA | Number of rows of input matrix A |
nA | Number of columns of input matrix A |
d_A | Left Input Matrix |
batch_A | Number of tiles in matrix A |
mB | Number of rows of input matrix B |
nB | Number of columns of input matrix B |
d_B | Right Input Matrix |
batch_B | Number of tiles in matrix B |
|
afError | af_kronProd_SS (float *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B) |
| Single Precision Kronecker Tensor Product - Real x Real.
|
afError | af_kronProd_SC (cuComplex *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B) |
| Single Precision Kronecker Tensor Product - Real x Complex.
|
afError | af_kronProd_CS (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B) |
| Single Precision Kronecker Tensor Product - Complex x Real.
|
afError | af_kronProd_CC (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B) |
| Single Precision Kronecker Tensor Product - Complex x Complex.
|
afError | af_kronProd_SD (float *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Single-precision Real x Double-precision Real.
|
afError | af_kronProd_SZ (cuComplex *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Single-precision Real x Double-precision Complex.
|
afError | af_kronProd_CD (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Single-precision Complex x Double-precision Real.
|
afError | af_kronProd_CZ (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Single-precision Complex x Double-precision Complex.
|
afError | af_kronProd_DS (float *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Double-precision Real x Single-precision Real.
|
afError | af_kronProd_DC (cuComplex *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Double-precision Real x Single-precision Complex.
|
afError | af_kronProd_ZS (cuComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Double-precision Complex x Single-precision Real.
|
afError | af_kronProd_ZC (cuComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B) |
| Kronecker Tensor Product - Double-precision Complex x Single-precision Complex.
|
afError | af_kronProd_DD (double *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B) |
| Double Precision Kronecker Tensor Product - Real x Complex.
|
afError | af_kronProd_DZ (cuDoubleComplex *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B) |
| Double Precision Kronecker Tensor Product - Real x Complex.
|
afError | af_kronProd_ZD (cuDoubleComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B) |
| Double Precision Kronecker Tensor Product - Complex x Real.
|
afError | af_kronProd_ZZ (cuDoubleComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B) |
| Double Precision Kronecker Tensor Product - Complex x Complex.
|