math.h File Reference

(2014-04-30 15:38:09 -0400, (build fd32605))

#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

Parameters
[in]ndims_outnumber of output dimensions
[in]dims_outoutput dimensions
[out]d_outThe output of the function
[in]ndims_innumber of input dimensions
[in]dims_ininput dimensions
[in]d_inThe input to the function
[in]orderThe order of the difference
[in]dimThe 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. 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.

Parameters
d_COutput Matrix
mANumber of rows of input matrix A
nANumber of columns of input matrix A
d_ALeft Input Matrix
batch_ANumber of tiles in matrix A
mBNumber of rows of input matrix B
nBNumber of columns of input matrix B
d_BRight Input Matrix
batch_BNumber 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. 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)