HMLP: High-performance Machine Learning Primitives
Main Page
Namespaces
Classes
Files
File List
conv_relu_pool2x2_6x8.hpp
1
#include <stdio.h>
2
#include <hmlp_internal.hpp>
3
4
// #define DEBUG_MICRO 1
5
6
void
conv_relu_pool2x2_d6x8
7
(
8
dim_t k,
9
double
* restrict alpha,
10
double
* restrict a,
11
double
* restrict b,
12
double
* restrict beta,
13
double
* restrict c, inc_t rs_c, inc_t cs_c,
14
aux_s<double, double, double, double>
*aux
15
);
16
17
18
19
struct
conv_relu_pool2x2_asm_d6x8
20
{
21
inline
void
operator()
22
(
23
int
k,
24
double
*a,
25
double
*b,
26
double
*c,
int
ldc,
27
aux_s<double, double, double, double>
*aux
28
)
const
29
{
30
double
alpha = 1.0;
31
double
beta = aux->pc ? 1.0 : 0.0;
32
conv_relu_pool2x2_d6x8
33
(
34
k,
35
&alpha,
36
a,
37
b,
38
&beta,
39
c, 1, ldc,
40
aux
41
);
42
};
// end inline void operator()
43
};
// end struct rank_k_asm_d8x4
conv_relu_pool2x2_asm_d6x8
Definition:
conv_relu_pool2x2_6x8.hpp:19
aux_s
Definition:
hmlp_internal.hpp:38
kernel
arm
armv8a
conv_relu_pool2x2_6x8.hpp
Generated by
1.8.11