XNOR-net on CPU AVX2

AlexeyAB

2018-08-07 0a326e7afe3e690c0b4cb64bbd0ce4f0603a7d85

 src/network.c

@@ -847,3 +847,25 @@
        }
    }
}



void calculate_binary_weights(network net)
{
    int j;
    for (j = 0; j < net.n; ++j) {
        layer *l = &net.layers[j];

        if (l->type == CONVOLUTIONAL) {
            //printf(" Merges Convolutional-%d and batch_norm \n", j);

            if (l->xnor) {
                //printf("\n %d \n", j);
                size_t ldb_align = 256; // 256bit for AVX2
                binary_transpose_align_weights(l, ldb_align);
            }
        }
    }
    //printf("\n calculate_binary_weights Done! \n");

}