HMLP: High-performance Machine Learning Primitives
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,
15 );
16 
17 
18 
20 {
21  inline void operator()
22  (
23  int k,
24  double *a,
25  double *b,
26  double *c, int ldc,
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
Definition: conv_relu_pool2x2_6x8.hpp:19
Definition: hmlp_internal.hpp:38