Searched defs:is_relu (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | l2_norm_fp32.c | 32 bool is_relu = param->act_type_ == ActType_Relu; in ThreadDivSqrtSum() local
53 bool is_relu = param->act_type_ == ActType_Relu; in ThreadTrailingAxis() local
|
| D | conv_depthwise_fp32.c | 180 … int in_kh_step, int in_kw_step, int kernel_w_step, bool is_relu, bool is_relu6) { in ConvDwBorderPixel()
264 int in_kh_step, int in_kw_step, bool is_relu, bool is_relu6) { in ConvDwCenter()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | quantize.c | 77 void CalculateActivationRangeQuantized(bool is_relu, bool is_relu6, int32_t zp, float scale, int *m… in CalculateActivationRangeQuantized()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 454 … int height, int width, int in_kh_step, int in_kw_step, int kernel_w_step, bool is_relu, in DepthwiseBorderPixelFp16()
522 … int in_sh_step, int in_sw_step, int in_kh_step, int in_kw_step, bool is_relu, bool is_relu6) { in DepthwiseCenterFp16()
|