blas.h
Go to the documentation of this file.
1 /*******************************************************
2  * Copyright (c) 2014, ArrayFire
3  * All rights reserved.
4  *
5  * This file is distributed under 3-clause BSD license.
6  * The complete license agreement can be obtained at:
7  * http://arrayfire.com/licenses/BSD-3-Clause
8  ********************************************************/
9 
17 #pragma once
18 
19 #include <af/defines.h>
20 
21 #ifdef __cplusplus
22 namespace af
23 {
24  class array;
52  AFAPI array matmul(const array &lhs, const array &rhs,
53  const matProp optLhs = AF_MAT_NONE,
54  const matProp optRhs = AF_MAT_NONE);
55 
69  AFAPI array matmulNT(const array &lhs, const array &rhs);
70 
84  AFAPI array matmulTN(const array &lhs, const array &rhs);
85 
99  AFAPI array matmulTT(const array &lhs, const array &rhs);
100 
116  AFAPI array matmul(const array &a, const array &b, const array &c);
117 
118 
135  AFAPI array matmul(const array &a, const array &b, const array &c, const array &d);
136 
137 
165  AFAPI array dot (const array &lhs, const array &rhs,
166  const matProp optLhs = AF_MAT_NONE,
167  const matProp optRhs = AF_MAT_NONE);
168 
179  AFAPI array transpose(const array& in, const bool conjugate = false);
180 
191  AFAPI void transposeInPlace(array& in, const bool conjugate = false);
192 }
193 #endif
194 
195 #ifdef __cplusplus
196 extern "C" {
197 #endif
198 
224  const af_array lhs, const af_array rhs,
225  const af_mat_prop optLhs, const af_mat_prop optRhs);
226 
227 
240  AFAPI af_err af_dot( af_array *out,
241  const af_array lhs, const af_array rhs,
242  const af_mat_prop optLhs, const af_mat_prop optRhs);
243 
256  AFAPI af_err af_transpose(af_array *out, af_array in, const bool conjugate);
257 
268  AFAPI af_err af_transpose_inplace(af_array in, const bool conjugate);
269 
270 
271 #ifdef __cplusplus
272 }
273 #endif
AFAPI af_err af_dot(af_array *out, const af_array lhs, const af_array rhs, const af_mat_prop optLhs, const af_mat_prop optRhs)
Scalar dot product between two vectors.
AFAPI af_err af_matmul(af_array *out, const af_array lhs, const af_array rhs, const af_mat_prop optLhs, const af_mat_prop optRhs)
Matrix multiply of two af_array.
Definition: algorithm.h:14
AFAPI array matmulTN(const array &lhs, const array &rhs)
Matrix multiply of two arrays.
Default.
Definition: defines.h:321
AFAPI af_err af_transpose(af_array *out, af_array in, const bool conjugate)
Transposes a matrix.
AFAPI array matmulNT(const array &lhs, const array &rhs)
Matrix multiply of two arrays.
af_err
Definition: defines.h:63
AFAPI array matmulTT(const array &lhs, const array &rhs)
Matrix multiply of two arrays.
AFAPI array matmul(const array &lhs, const array &rhs, const matProp optLhs=AF_MAT_NONE, const matProp optRhs=AF_MAT_NONE)
Matrix multiply of two arrays.
af_mat_prop matProp
Definition: defines.h:472
#define AFAPI
Definition: defines.h:31
AFAPI array dot(const array &lhs, const array &rhs, const matProp optLhs=AF_MAT_NONE, const matProp optRhs=AF_MAT_NONE)
Dot Product.
AFAPI af_err af_transpose_inplace(af_array in, const bool conjugate)
Transposes a matrix in-place.
static af::array array(af::dim4 idims, cl_mem buf, af::dtype type, bool retain=false)
Create an af::array object from an OpenCL cl_mem buffer.
Definition: opencl.h:327
af_mat_prop
Definition: defines.h:320
void * af_array
Definition: defines.h:222
AFAPI void transposeInPlace(array &in, const bool conjugate=false)
Transposes a matrix in-place.
AFAPI array transpose(const array &in, const bool conjugate=false)
Transposes a matrix.