Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dactivation_grad.c49 float16x8_t zero_8 = vdupq_n_f16(0); in Relu6Fp16Grad() local
54 float16x8_t max_8 = vmaxq_f16(src1_8, zero_8); in Relu6Fp16Grad()
57 float16x8_t dst_8 = vbslq_f16(mask_8, src0_8, zero_8); in Relu6Fp16Grad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dactivation_fp32.c24 MS_FLOAT32X8 zero_8 = MS_MOV256_F32(0.0f); in Fp32Relu() local
26 MS_ST256_F32(dst + i, MS_MAX256_F32(MS_LD256_F32(src + i), zero_8)); in Fp32Relu()
46 MS_FLOAT32X8 zero_8 = MS_MOV256_F32(0.0f); in Fp32Relu6() local
49 MS_FLOAT32X8 dst_tmp = MS_MAX256_F32(MS_LD256_F32(src + i), zero_8); in Fp32Relu6()