Home
last modified time | relevance | path

Searched defs:offset_8 (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dscale_fp16.c31 float16x8_t offset_8 = vdupq_n_f16(offset[i]); in Fp16ScaleInner() local
55 float16x8_t offset_8 = vld1q_f16(offset + index); in Fp16ScaleAxis() local
97 float16x8_t offset_8 = vdupq_n_f16(offset[i]); in Fp16ScaleInnerRelu() local
125 float16x8_t offset_8 = vld1q_f16(offset + index); in Fp16ScaleAxisRelu() local
170 float16x8_t offset_8 = vdupq_n_f16(offset[i]); in Fp16ScaleInnerRelu6() local
199 float16x8_t offset_8 = vld1q_f16(offset + index); in Fp16ScaleAxisRelu6() local
Dbatchnorm_fp16.c66 MS_FLOAT16X8 offset_8 = MS_LDQ_F16(offset + c); in FusedBatchNormFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dscale_fp32.c29 MS_FLOAT32X8 offset_8 = MS_MOV256_F32(offset[i]); in ScaleInner() local
64 MS_FLOAT32X8 offset_8 = MS_LD256_F32(offset + index); in ScaleAxis() local
117 MS_FLOAT32X8 offset_8 = MS_MOV256_F32(offset[i]); in ScaleInnerRelu() local
161 MS_FLOAT32X8 offset_8 = MS_LD256_F32(offset + index); in ScaleAxisRelu() local
219 MS_FLOAT32X8 offset_8 = MS_MOV256_F32(offset[i]); in ScaleInnerRelu6() local
266 MS_FLOAT32X8 offset_8 = MS_LD256_F32(offset + index); in ScaleAxisRelu6() local