Searched defs:weight_mask (Results 1 – 3 of 3) sorted by relevance
123 __m256i *weight_mask) { in load_shuffle_masks_for_2x2_convolve()139 __m256 *shuffle_weight, __m256i *weight_mask) { in prepare_weights_for_2x2_convolve()370 __m256i weight_mask[2]; in cnn_convolve_no_maxpool_padding_valid_layer1_avx2() local417 __m256i weight_mask[2]; in cnn_convolve_no_maxpool_padding_valid_layer2_avx2() local
75 alignas(kMaxAlignment) uint8_t weight_mask[kMaxSuperBlockSizeSquareInPixels]; member
900 WeightMaskFuncs weight_mask; member