Home
last modified time | relevance | path

Searched defs:scale_8 (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dscale_fp16.c30 float16x8_t scale_8 = vdupq_n_f16(scale[i]); in Fp16ScaleInner() local
54 float16x8_t scale_8 = vld1q_f16(scale + index); in Fp16ScaleAxis() local
96 float16x8_t scale_8 = vdupq_n_f16(scale[i]); in Fp16ScaleInnerRelu() local
124 float16x8_t scale_8 = vld1q_f16(scale + index); in Fp16ScaleAxisRelu() local
169 float16x8_t scale_8 = vdupq_n_f16(scale[i]); in Fp16ScaleInnerRelu6() local
198 float16x8_t scale_8 = vld1q_f16(scale + index); in Fp16ScaleAxisRelu6() local
Dpower_fp16.c74 float16x8_t scale_8 = vmovq_n_f16(scale); in PowerBroadCastFp16() local
92 float16x8_t scale_8 = vmovq_n_f16(scale); in PowerSingleFp16() local
Dbatchnorm_fp16.c65 MS_FLOAT16X8 scale_8 = MS_LDQ_F16(scale + c); in FusedBatchNormFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dscale_fp32.c28 MS_FLOAT32X8 scale_8 = MS_MOV256_F32(scale[i]); in ScaleInner() local
63 MS_FLOAT32X8 scale_8 = MS_LD256_F32(scale + index); in ScaleAxis() local
116 MS_FLOAT32X8 scale_8 = MS_MOV256_F32(scale[i]); in ScaleInnerRelu() local
160 MS_FLOAT32X8 scale_8 = MS_LD256_F32(scale + index); in ScaleAxisRelu() local
218 MS_FLOAT32X8 scale_8 = MS_MOV256_F32(scale[i]); in ScaleInnerRelu6() local
265 MS_FLOAT32X8 scale_8 = MS_LD256_F32(scale + index); in ScaleAxisRelu6() local