Optimized on CPU: gemm_bin, im2col, activation, transpose

This commit is contained in:
AlexeyAB
2018-08-08 19:28:39 +03:00
parent a284a7da8d
commit d6162af210
6 changed files with 450 additions and 126 deletions

View File

@ -862,9 +862,8 @@ void calculate_binary_weights(network net)
if (l->xnor) {
//printf("\n %d \n", j);
size_t ldb_align = 256; // 256bit for AVX2
if (l->size*l->size*l->c > 4096) ldb_align = 4096;
binary_transpose_align_weights(l, ldb_align);
binary_align_weights(l, ldb_align);
}
}
}