Documentation
|
|
#include "defines.h"Go to the source code of this file.
Namespaces | |
| af | |
Functions | |||||||||||||||||||||||||
| array | diff1 (const array &in, int dim=-1) | ||||||||||||||||||||||||
| array | diff2 (const array &in, int dim=-1) | ||||||||||||||||||||||||
| void | grad (array &dcols, array &drows, const array &in) | ||||||||||||||||||||||||
Device pointer interface: Difference along a dimension | |||||||||||||||||||||||||
Difference function
| |||||||||||||||||||||||||
| 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. More... | |||||||||||||||||||||||||
| 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. More... | |||||||||||||||||||||||||
| afError | af_grad_B (bool *d_dx, bool *d_dy, unsigned nx, unsigned ny, const bool *d_in, unsigned batch) | ||||||||||||||||||||||||
| afError | af_grad_U (unsigned *d_dx, unsigned *d_dy, unsigned nx, unsigned ny, const unsigned *d_in, unsigned batch) | ||||||||||||||||||||||||
| Unsigned integer gradient function. More... | |||||||||||||||||||||||||
| afError | af_grad_I (int *d_dx, int *d_dy, unsigned nx, unsigned ny, const int *d_in, unsigned batch) | ||||||||||||||||||||||||
| Integer gradient function. More... | |||||||||||||||||||||||||
| 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 function. More... | |||||||||||||||||||||||||
| 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 function. More... | |||||||||||||||||||||||||
Device pointer interface: Kronecker product | |||||||||||||||||||||||||
Kronecker Tensor Product: Computes Kronecker product of matrices A, 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. More... | |||||||||||||||||||||||||
| afError | af_kronProd_SC (cfloat *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cfloat *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Single Precision Kronecker Tensor Product - Real x Complex. More... | |||||||||||||||||||||||||
| afError | af_kronProd_CS (cfloat *d_C, unsigned mA, unsigned nA, const cfloat *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Single Precision Kronecker Tensor Product - Complex x Real. More... | |||||||||||||||||||||||||
| afError | af_kronProd_CC (cfloat *d_C, unsigned mA, unsigned nA, const cfloat *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cfloat *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Single Precision Kronecker Tensor Product - Complex x Complex. More... | |||||||||||||||||||||||||
| 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. More... | |||||||||||||||||||||||||
| afError | af_kronProd_SZ (cfloat *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cdouble *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Kronecker Tensor Product - Single-precision Real x Double-precision Complex. More... | |||||||||||||||||||||||||
| afError | af_kronProd_CD (cfloat *d_C, unsigned mA, unsigned nA, const cfloat *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. More... | |||||||||||||||||||||||||
| afError | af_kronProd_CZ (cfloat *d_C, unsigned mA, unsigned nA, const cfloat *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cdouble *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Kronecker Tensor Product - Single-precision Complex x Double-precision Complex. More... | |||||||||||||||||||||||||
| 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. More... | |||||||||||||||||||||||||
| afError | af_kronProd_DC (cfloat *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cfloat *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Kronecker Tensor Product - Double-precision Real x Single-precision Complex. More... | |||||||||||||||||||||||||
| afError | af_kronProd_ZS (cfloat *d_C, unsigned mA, unsigned nA, const cdouble *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. More... | |||||||||||||||||||||||||
| afError | af_kronProd_ZC (cfloat *d_C, unsigned mA, unsigned nA, const cdouble *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cfloat *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Kronecker Tensor Product - Double-precision Complex x Single-precision Complex. More... | |||||||||||||||||||||||||
| 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. More... | |||||||||||||||||||||||||
| afError | af_kronProd_DZ (cdouble *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cdouble *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Double Precision Kronecker Tensor Product - Real x Complex. More... | |||||||||||||||||||||||||
| afError | af_kronProd_ZD (cdouble *d_C, unsigned mA, unsigned nA, const cdouble *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Double Precision Kronecker Tensor Product - Complex x Real. More... | |||||||||||||||||||||||||
| afError | af_kronProd_ZZ (cdouble *d_C, unsigned mA, unsigned nA, const cdouble *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cdouble *d_B, unsigned batch_B) | ||||||||||||||||||||||||
| Double Precision Kronecker Tensor Product - Complex x Complex. More... | |||||||||||||||||||||||||
| 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 (cfloat *d_out, unsigned n, const cfloat *d_in, float nu) | ||||||||||||||||||||||||
| afError | af_besselj_D (double *d_out, unsigned n, const double *d_in, double nu) | ||||||||||||||||||||||||
| afError | af_besselj_Z (cdouble *d_out, unsigned n, const cdouble *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) | ||||||||||||||||||||||||