fool/math_util.hpp at master · DaFuCoding/fool · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#ifndef MATHUTIL_HPP
#define MATHUTIL_HPP
#include "common.hpp"
#include "cblas.h"
#include "Eigen/Core"
#include "Eigen/Dense"

namespace fool {
template<typename Dtype>
void fool_set(const int N, const Dtype value, Dtype* Y);

namespace math{
// Common Eigen types that we will often use
template <typename T>
using EigenMatrixMap =
		Eigen::Map<Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic> >;
template <typename T>
using EigenArrayMap =
		Eigen::Map<Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic> >;
template <typename T>
using EigenVectorMap = Eigen::Map<Eigen::Matrix<T, Eigen::Dynamic, 1> >;
template <typename T>
using EigenVectorArrayMap = Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1> >;
template <typename T>
using ConstEigenMatrixMap =
		Eigen::Map<const Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic> >;
template <typename T>
using ConstEigenArrayMap =
		Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic> >;
template <typename T>
using ConstEigenVectorMap =
		Eigen::Map<const Eigen::Matrix<T, Eigen::Dynamic, 1> >;
template <typename T>
using ConstEigenVectorArrayMap =
		Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, 1> >;

template<typename Dtype>
void Gemm(const CBLAS_TRANSPOSE TransA,
					const CBLAS_TRANSPOSE TransB,
					const int M,
					const int N,
					const int K,
					const float alpha,
					const float* A,
					const float* B,
					const float beta,
					float* C);
template<typename Dtype>
void Gemv(const CBLAS_TRANSPOSE TransA, const int M,
					const int N, const float alpha, const float* A, const float* x,
					const float beta, float* y);

template<typename Dtype>
void Axpy();
} // math namespace

}
#endif // MATHUTIL_HPP