Home
last modified time | relevance | path

Searched defs:relu6 (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/
Dsse_common.h20 static inline void ActBlock1(__m128 *v1, size_t relu, size_t relu6) { in ActBlock1()
31 static inline void ActBlock2(__m128 *v1, __m128 *v2, size_t relu, size_t relu6) { in ActBlock2()
44 … inline void ActBlock4(__m128 *v1, __m128 *v2, __m128 *v3, __m128 *v4, size_t relu, size_t relu6) { in ActBlock4()
63 __m128 relu6 = _mm_set_ps1(6.0); in ActBlock8() local
DConvDwFp32IndirectRow.c28 size_t output_width, size_t input_stride, size_t relu, size_t relu6) { in ConvDwFp32Avx5x5()
DDepthwiseFp32_Sse.c24 … size_t in_kh_step, size_t in_kw_step, size_t kernel_w_step, size_t relu, size_t relu6) { in ConvDwFp32Border()
112 … size_t in_sw_step, size_t in_kh_step, size_t in_kw_step, size_t relu, size_t relu6) { in ConvDwFp32Center()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dcommon_func_fp32.h35 size_t relu6; member
Dactivation_fp32.c192 float relu6 = MSMIN(MSMAX(in + 3, 0), 6); in HSwish() local
200 float relu6 = MSMIN(MSMAX(src[i] + 3, 0), 6); in HSigmoid() local
Dconv_depthwise_fp32.c44 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDw() local
212 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDwBorder() local
308 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDwSWFp32() local
577 bool relu, bool relu6) { in ConvDw3x3Line()
680 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDw3x3() local
770 int output_width, int input_stride, bool relu, bool relu6, int kernel) { in ConvDwFp32IndirectRow()
808 int output_width, int input_stride, bool relu, bool relu6, int kernel) { in ConvDwFp32IndirectRow()
821 int output_width, int input_stride, bool relu, bool relu6, int kernel) { in ConvDwFp32IndirectRow()
841 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDwIndirection() local
964 bool relu6 = conv_param->act_type_ == ActType_Relu6; in DeconvDwPost() local
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc123 auto relu6 = cast->cast<CNodePtr>()->input(1); in TEST_F() local
213 auto relu6 = cast->cast<CNodePtr>()->input(1); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_depthwise_fp16.c250 int width, int ori_channel, bool relu, bool relu6) { in ConvDw3x3LineFp16()
356 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDw3x3Fp16() local
405 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDwFp16() local
488 bool relu6 = conv_param->act_type_ == ActType_Relu6; in DepthwiseBorderFp16() local
577 bool relu6 = conv_param->act_type_ == ActType_Relu6; in ConvDwC8Fp16() local
722 bool relu6 = conv_param->act_type_ == ActType_Relu6; in DeconvDepthwisePostFuncFp16() local
Dactivation_fp16.c156 float16_t relu6 = MSMIN(MSMAX(in + 3.0f, 0.0f), 6.0f); in HSwishFp16() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_depthwise_slidewindow_int8.cc245 bool relu6 = conv_param_->act_type_ == ActType_Relu6; in ReinitQuantParam() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dconvolution_base.cc411 bool relu6 = conv_param_->act_type_ == ActType_Relu6; in SetQuantParam() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/
Dconv2d_base_coder.cc324 bool relu6 = conv_param_->act_type_ == ActType_Relu6; in SetQuantParam() local