Searched defs:shuffle_weight (Results 1 – 1 of 1) sorted by relevance
68 const int cstep, __m256 *shuffle_weight, const __m256i weight_mask_0, in prepare_weights_for_5x5_convolve()139 __m256 *shuffle_weight, __m256i *weight_mask) { in prepare_weights_for_2x2_convolve()282 __m256 shuffle_weight[10]; in cnn_convolve_no_maxpool_padding_valid_5x5_avx2() local390 __m256 shuffle_weight[2]; in cnn_convolve_no_maxpool_padding_valid_layer1_avx2() local441 __m256 shuffle_weight[2]; in cnn_convolve_no_maxpool_padding_valid_layer2_avx2() local