Home
last modified time | relevance | path

Searched refs:relu_type (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dcommon_func_fp32.c20 … 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()
Dcommon_func_fp32.h43 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/
DPostFuncBiasReluC4.c23 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()
DPostFuncBiasReluC8.c23 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()
Dsse_common.h62 __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/
Dcommon_func_fp16.h29 size_t plane_size, size_t stride, size_t relu_type);
35 size_t plane_size, size_t plane_stride, size_t relu_type);
Dcommon_func_fp16.c68 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