Home
last modified time | relevance | path

Searched refs:oc_tmp (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c179 for (int oc_tmp = 0; oc_tmp < oc_block; oc_tmp += C8NUM) { in PackNHWCToNXHWCXFp32() local
182 … Transpose8X8Fp32Avx(src + ic, tmp_weight + ic * oc_block + oc_tmp, input_channel, oc_block); in PackNHWCToNXHWCXFp32()
186 tmp_weight[ic * oc_block + oc_tmp + j] = src[ic + input_channel * j]; in PackNHWCToNXHWCXFp32()
202 for (int oc_tmp = 0; oc_tmp < oc_block; oc_tmp += C8NUM) { in PackNHWCToNXHWCXFp32() local
207 tmp_weight + hw * oc_block * input_channel + ic * oc_block + oc_tmp, in PackNHWCToNXHWCXFp32()
212 tmp_weight[ic * oc_block + oc_tmp + j + hw * oc_block * input_channel] = in PackNHWCToNXHWCXFp32()