/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
D | common_func_fp32.c | 20 … size_t plane_size, size_t plane_stride, size_t oc_stride, ActType relu_type, int size) { in PostConvFuncComm() argument 34 … value = (relu_type == ActType_Relu || relu_type == ActType_Relu6) ? (MSMAX(0.f, value)) : (value); in PostConvFuncComm() 35 value = (relu_type == ActType_Relu6) ? (MSMIN(6.f, value)) : (value); in PostConvFuncComm() 43 size_t plane_size, size_t stride, size_t relu_type) { in PostConvFuncFp32C8() argument 45 …m(c8_out_ptr, out_ptr, bias_ptr, output_channel, plane_size, plane_size, stride, relu_type, C8NUM); in PostConvFuncFp32C8() 50 …tFuncBiasReluC8(out_ptr, c8_out_ptr, bias_ptr, oc8div, oc8mod, plane_size, stride_size, relu_type); in PostConvFuncFp32C8() 56 size_t plane_size, size_t plane_stride, size_t relu_type) { in PostConvFuncFp32C4() argument 61 …tFuncBiasReluC4(out_ptr, c4_out_ptr, bias_ptr, oc4div, oc4mod, plane_size, stride_size, relu_type); in PostConvFuncFp32C4() 63 …c4_out_ptr, out_ptr, bias_ptr, output_channel, plane_size, plane_stride, output_channel, relu_type, in PostConvFuncFp32C4()
|
D | common_func_fp32.h | 43 size_t plane_size, size_t stride, size_t relu_type); 45 size_t plane_size, size_t plane_stride, size_t relu_type); 64 size_t plane_size, size_t stride, size_t relu_type); 68 size_t plane_size, size_t plane_stride, size_t relu_type);
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/ |
D | PostFuncBiasReluC4.c | 23 size_t plane_size, size_t plane_stride, size_t relu_type) { in PostFuncBiasReluC4() argument 45 ActBlock4(&src1, &src2, &src3, &src4, relu_type == 1, relu_type == 3); in PostFuncBiasReluC4() 60 ActBlock1(&src1, relu_type == 1, relu_type == 3); in PostFuncBiasReluC4() 82 ActBlock1(&src1, relu_type == 1, relu_type == 3); in PostFuncBiasReluC4()
|
D | PostFuncBiasReluC8.c | 23 size_t plane_size, size_t stride, size_t relu_type) { in PostFuncBiasReluC8() argument 52 ActBlock8(&src1, &src2, &src3, &src4, &src5, &src6, &src7, &src8, relu_type); in PostFuncBiasReluC8() 73 ActBlock2(&src1, &src2, relu_type == 1, relu_type == 3); in PostFuncBiasReluC8() 96 ActBlock2(&src1, &src2, relu_type == 1, relu_type == 3); in PostFuncBiasReluC8()
|
D | sse_common.h | 62 __m128 *v8, size_t relu_type) { in ActBlock8() argument 65 switch (relu_type) { in ActBlock8()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
D | common_func_fp16.h | 29 size_t plane_size, size_t stride, size_t relu_type); 35 size_t plane_size, size_t plane_stride, size_t relu_type);
|
D | common_func_fp16.c | 68 size_t plane_size, size_t plane_stride, size_t relu_type) { in PostFuncBiasReluC4Fp16() argument 73 size_t plane_size, size_t stride, size_t relu_type) { in PostFuncBiasReluC8Fp16() argument
|