Searched refs:offset_8 (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | scale_fp16.c | 31 float16x8_t offset_8 = vdupq_n_f16(offset[i]); in Fp16ScaleInner() local
32 float16x8_t result = vfmaq_f16(offset_8, data, scale_8); in Fp16ScaleInner()
55 float16x8_t offset_8 = vld1q_f16(offset + index); in Fp16ScaleAxis() local
56 float16x8_t result = vfmaq_f16(offset_8, data, scale_8); in Fp16ScaleAxis()
97 float16x8_t offset_8 = vdupq_n_f16(offset[i]); in Fp16ScaleInnerRelu() local
98 float16x8_t tmp = vfmaq_f16(offset_8, data, scale_8); in Fp16ScaleInnerRelu()
125 float16x8_t offset_8 = vld1q_f16(offset + index); in Fp16ScaleAxisRelu() local
126 float16x8_t tmp = vfmaq_f16(offset_8, data, scale_8); in Fp16ScaleAxisRelu()
170 float16x8_t offset_8 = vdupq_n_f16(offset[i]); in Fp16ScaleInnerRelu6() local
171 float16x8_t tmp = vfmaq_f16(offset_8, data, scale_8); in Fp16ScaleInnerRelu6()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | scale_fp32.c | 30 MS_FLOAT32X8 offset_8 = MS_MOV256_F32(offset[i]); in ScaleInner() local
34 MS_FLOAT32X8 result = MS_MLA256_F32(offset_8, data, scale_8); in ScaleInner()
65 MS_FLOAT32X8 offset_8 = MS_LD256_F32(offset + index); in ScaleAxis() local
67 MS_FLOAT32X8 result = MS_MLA256_F32(offset_8, data, scale_8); in ScaleAxis()
120 MS_FLOAT32X8 offset_8 = MS_MOV256_F32(offset[i]); in ScaleInnerRelu() local
124 MS_FLOAT32X8 tmp = MS_MLA256_F32(offset_8, data, scale_8); in ScaleInnerRelu()
164 MS_FLOAT32X8 offset_8 = MS_LD256_F32(offset + index); in ScaleAxisRelu() local
166 MS_FLOAT32X8 tmp = MS_MLA256_F32(offset_8, data, scale_8); in ScaleAxisRelu()
224 MS_FLOAT32X8 offset_8 = MS_MOV256_F32(offset[i]); in ScaleInnerRelu6() local
228 MS_FLOAT32X8 tmp = MS_MLA256_F32(offset_8, data, scale_8); in ScaleInnerRelu6()
[all …]
|