|
|
| template<typename ty > |
| ty | norm (const array &in, float p=af::NaN) |
| | Matrix or vector norm. More...
|
| |
|
| template<typename ty > |
| ty | dot (const array &x, const array &y) |
| | Scalar dot product between two vectors. More...
|
| |
|
| array | matmul (const array &, const array &, af_matprop_t=af_none, af_matprop_t=af_none) |
| | Multiply two arrays together. More...
|
| |
| array | matmul (const array &, const array &, const array &) |
| | Multiply three arrays together. More...
|
| |
| array | matmul (const array &, const array &, const array &, const array &) |
| | Multiply four arrays together. More...
|
| |
|
| afError | af_gemm_S (char TRANSA, char TRANSB, int M, int N, int K, float ALPHA, const float *d_A, int LDA, const float *d_B, int LDB, float BETA, float *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_gemm_D (char TRANSA, char TRANSB, int M, int N, int K, double ALPHA, const double *d_A, int LDA, const double *d_B, int LDB, double BETA, double *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0) |
| | Double precision, Real matrix multiply. More...
|
| |
| afError | af_gemm_C (char TRANSA, char TRANSB, int M, int N, int K, cfloat ALPHA, const cfloat *d_A, int LDA, const cfloat *d_B, int LDB, cfloat BETA, cfloat *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0) |
| | Single precision, Complex matrix multiply. More...
|
| |
| afError | af_gemm_Z (char TRANSA, char TRANSB, int M, int N, int K, cdouble ALPHA, const cdouble *d_A, int LDA, const cdouble *d_B, int LDB, cdouble BETA, cdouble *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0) |
| | Double precision, Complex matrix multiply. More...
|
| |
| afError | af_matmul_SS (const float *d_A, const float *d_B, float *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| | Mixed precision, mixed complexity matrix multiply. More...
|
| |
| afError | af_matmul_DS (const double *d_A, const float *d_B, float *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_SD (const float *d_A, const double *d_B, float *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_DD (const double *d_A, const double *d_B, double *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_CS (const cfloat *d_A, const float *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_ZS (const cdouble *d_A, const float *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_CD (const cfloat *d_A, const double *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_ZD (const cdouble *d_A, const double *d_B, cdouble *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_SC (const float *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_DC (const double *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_SZ (const float *d_A, const cdouble *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_DZ (const double *d_A, const cdouble *d_B, cdouble *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_CC (const cfloat *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_ZC (const cdouble *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_CZ (const cfloat *d_A, const cdouble *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
| afError | af_matmul_ZZ (const cdouble *d_A, const cdouble *d_B, cdouble *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0) |
| |
|
| afError | af_norm_vector_S (float *h_dst, unsigned numel, const float *d_src) |
| |
| afError | af_norm_vector_D (double *h_dst, unsigned numel, const double *d_src) |
| | norm The value of a double precision vector More...
|
| |
| afError | af_norm_S (float *d_dst, unsigned ndims, const unsigned *dims, const float *d_src, int dim) |
| | norm the value along a dimension in single precision matrix More...
|
| |
| afError | af_norm_D (double *d_dst, unsigned ndims, const unsigned *dims, const double *d_src, int dim) |
| | norm the value along a dimension in double precision matrix More...
|
| |
|
| afError | af_dot_S (float *h_res, unsigned len, const float *d_A, const float *d_B) |
| |
| afError | af_dot_D (double *h_res, unsigned len, const double *d_A, const double *d_B) |
| |
| afError | af_dot_C (cfloat *h_res, unsigned len, const cfloat *d_A, const cfloat *d_B) |
| |
| afError | af_dot_Z (cdouble *h_res, unsigned len, const cdouble *d_A, const cdouble *d_B) |
| |
