Searched refs:src_sw (Results 1 – 2 of 2) sorted by relevance
1563 const float *src_sw[12]; in Conv1x1SWOWxOCKernel() local1584 src_sw[i] = src + i * in_sw_step; in Conv1x1SWOWxOCKernel()1593 dst_data[i * oc_block + j] += src_sw[i][ic] * weight_data[j]; in Conv1x1SWOWxOCKernel()
1478 const float *src_sw[12]; in MatVecMulRowxColKernel() local1493 src_sw[i] = src + i * deep; in MatVecMulRowxColKernel()1503 … _mm256_fmadd_ps(_mm256_set1_ps(src_sw[i][ic]), weight_data[j], dst_data[i * col_block + j]); in MatVecMulRowxColKernel()