mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

fully separate C-API from CPP-API

Browse Source
This commit is contained in:
Stefano Sinigardi
2019-02-19 15:57:18 +01:00
parent 6e7c7976d7
commit 00de023601
8 changed files with 66 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/convolutional_layer.c
View File

@ -18,6 +18,12 @@
#include "xnor_layer.h"
#endif
#ifdef __cplusplus
#define PUT_IN_REGISTER
#else
#define PUT_IN_REGISTER register
#endif
#ifndef AI2
#define AI2 0
void forward_xnor_layer(layer l, network_state state);
@ -644,7 +650,7 @@ void gemm_nn_custom(int M, int N, int K, float ALPHA,
int i, j, k;
for (i = 0; i < M; ++i) {
for (k = 0; k < K; ++k) {
float A_PART = ALPHA * A[i * lda + k];
PUT_IN_REGISTER float A_PART = ALPHA * A[i * lda + k];
//printf("\n weight = %f \n", A_PART);
for (j = 0; j < N; ++j) {
C[i*ldc + j] += A_PART*B[k*ldb + j];