Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c136 int c4_block_num = i / C4NUM; in PackWeightToC4Fp16() local
139 int dst_ic_offset = dst_oc_offset + c4_block_num * kernel_plane * C4NUM + c4_block_rem; in PackWeightToC4Fp16()
155 int c4_block_num = i / C4NUM; in PackNHWCToNC4HW4Fp16() local
158 int dst_ic_offset = dst_kernel_offset + c4_block_num * plane * C4NUM + c4_block_rem; in PackNHWCToNC4HW4Fp16()
171 int c4_block_num = c / C4NUM; in PackNCHWToNC4HW4Fp16() local
174 int dst_c_offset = dst_offset + c4_block_num * plane * C4NUM; in PackNCHWToNC4HW4Fp16()
349 int c4_block_num = c / C4NUM; in PackNC4HW4ToNHWC4Fp16() local
351 int src_c_offset = src_offset + c4_block_num * plane * C4NUM + c4_block_res; in PackNC4HW4ToNHWC4Fp16()
352 int dst_c_offset = dst_offset + c4_block_num * C4NUM + c4_block_res; in PackNC4HW4ToNHWC4Fp16()
368 int c4_block_num = c / C4NUM; in PackNC4HW4ToNHWCFp16() local
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/
Dutils.h75 int c4_block_num = c / C4NUM; in PackNCHWToNC4HW4() local
78 int dst_c_offset = dst_offset + c4_block_num * plane_out; in PackNCHWToNC4HW4()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c308 int c4_block_num = c / C4NUM; in PackNC4HW4ToNHWC4Fp32() local
310 int src_c_offset = src_offset + c4_block_num * plane * C4NUM + c4_block_res; in PackNC4HW4ToNHWC4Fp32()
311 int dst_c_offset = dst_offset + c4_block_num * C4NUM + c4_block_res; in PackNC4HW4ToNHWC4Fp32()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpack_int8.c991 int c4_block_num = c / C4NUM; in PackDeconvDepthwiseInt8Weight() local
994 int16_t *dst_c = packed_weight_ + c4_block_num * plane * C4NUM; in PackDeconvDepthwiseInt8Weight()