blas.h

Go to the documentation of this file.

/*******************************************************
 * Copyright (c) 2014, ArrayFire
 * All rights reserved.
 *
 * This file is distributed under 3-clause BSD license.
 * The complete license agreement can be obtained at:
 * http://arrayfire.com/licenses/BSD-3-Clause
 ********************************************************/

#pragma once

#include <af/defines.h>

#ifdef __cplusplus
namespace af
{
    class array;
    AFAPI array matmul(const array &lhs, const array &rhs,
                       const matProp optLhs = AF_MAT_NONE,
                       const matProp optRhs = AF_MAT_NONE);

    AFAPI array matmulNT(const array &lhs, const array &rhs);

    AFAPI array matmulTN(const array &lhs, const array &rhs);

    AFAPI array matmulTT(const array &lhs, const array &rhs);

    AFAPI array matmul(const array &a, const array &b, const array &c);


    AFAPI array matmul(const array &a, const array &b, const array &c, const array &d);


    AFAPI array dot   (const array &lhs, const array &rhs,
                       const matProp optLhs = AF_MAT_NONE,
                       const matProp optRhs = AF_MAT_NONE);

    AFAPI array transpose(const array& in, const bool conjugate = false);

    AFAPI void transposeInPlace(array& in, const bool conjugate = false);
}
#endif

#ifdef __cplusplus
extern "C" {
#endif

    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);


    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);

    AFAPI af_err af_transpose(af_array *out, af_array in, const bool conjugate);

    AFAPI af_err af_transpose_inplace(af_array in, const bool conjugate);


#ifdef __cplusplus
}
#endif