Home
last modified time | relevance | path

Searched defs:act_type (Results 1 – 25 of 33) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dmatmul_fp16.h31 #define DO_RELU(value, act_type) \ argument
34 #define DO_RELU6(value, act_type) \ argument
Dcommon_func_fp16.c21 ActType act_type, int size) { in PostConvFuncCommFp16()
43 size_t oc_stride, ActType act_type) { in PostConvFuncFp16C8()
55 size_t plane_stride, ActType act_type) { in PostConvFuncFp16C4()
Dmatmul_fp16.c229 …16(const float16_t *a, const float16_t *b, float16_t *dst, const float16_t *bias, ActType act_type, in MatMul16x8Fp16()
291 …16(const float16_t *a, const float16_t *b, float16_t *dst, const float16_t *bias, ActType act_type, in MatMul12x8Fp16()
354 …16(const float16_t *a, const float16_t *b, float16_t *dst, const float16_t *bias, ActType act_type, in MatMul12x16Fp16()
376 …Fp16(const float16_t *a, const float16_t *b, float16_t *c, const float16_t *bias, ActType act_type, in MatMulFp16()
398 …nst float16_t *a, const float16_t *b, float16_t *c, const float16_t *bias, int act_type, int depth, in VecMatmulFp16()
473 …lA32Fp16(const float16_t *a, const float16_t *b, float16_t *c, const float16_t *bias, int act_type, in MatVecMulA32Fp16()
488 …Fp16(const float16_t *a, const float16_t *b, float16_t *c, const float16_t *bias, ActType act_type, in MatVecMulFp16()
926 …BaseFp16(const float16_t *a, const float16_t *b, float16_t *c, const float16_t *bias, int act_type, in MatmulBaseFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dmatmul_fp32.h29 #define DO_RELU(value, act_type) \ argument
32 #define DO_RELU6(value, act_type) \ argument
Dadder_fp32.c22 void Adder12x4(const float *a, const float *b, float *dst, const float *bias, ActType act_type, int… in Adder12x4()
44 void AdderOpt(const float *a, const float *b, float *c, const float *bias, ActType act_type, int de… in AdderOpt()
Dmatmul_fp32.c877 void MatVecMulFp32(const float *a, const float *b, float *c, const float *bias, int act_type, int d… in MatVecMulFp32()
893 void MatVecMulFp32Neon64(const float *a, const float *b, float *c, const float *bias, int act_type,… in MatVecMulFp32Neon64()
984 void MatMul12x8(const float *a, const float *b, float *dst, const float *bias, ActType act_type, in… in MatMul12x8()
1046 void MatMulOpt(const float *a, const float *b, float *c, const float *bias, ActType act_type, int d… in MatMulOpt()
1072 void MatVecMulAvxFp32(const float *a, const float *b, float *c, const float *bias, int act_type, in… in MatVecMulAvxFp32()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dl2norm_fp32_test.cc90 ActType act_type = ActType_No; in TEST_F() local
111 ActType act_type = ActType_Relu; in TEST_F() local
132 ActType act_type = ActType_No; in TEST_F() local
153 ActType act_type = ActType_Relu6; in TEST_F() local
Dscale_fp32_tests.cc67 ActivationType act_type, const int thread_num) { in Prepare()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Ddepthwise_conv2d_tests.cc27 … int pad_r, int dilation_h, int dilation_w, ActType act_type, int input_channel) { in CreateParameter()
58 ActType act_type = ActType_No; in TEST_F() local
104 ActType act_type = ActType_No; in TEST_F() local
151 ActType act_type = ActType_No; in TEST_F() local
195 ActType act_type = ActType_No; in TEST_F() local
Dconv2d_tests.cc25 ConvParameter *CreateParameter(const std::string &attr, ActType act_type) { in CreateParameter()
40 float *output_data, ActType act_type, bool fp16_enable, float atol = 1e-9) { in TestMain_Conv2D()
267 auto act_type = std::get<6>(case_); in TEST_F() local
Dactivation_tests.cc25 OpParameter *CreateParameter(schema::ActivationType act_type) { in CreateParameter()
Dpooling_tests.cc27 ActType act_type = ActType_No) { in CreateParameter()
/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Ddepthwise_conv2d_populate_v0.cc66 auto act_type = depthwise_conv2d_prim->activationType(); in PopulateConvDwParameter() local
Ddeconv2d_populate_v0.cc52 auto act_type = deconv2d_prim->activationType(); in PopulateDeconvParameter() local
Dconv2d_populate_v0.cc68 auto act_type = conv2d_prim->activationType(); in PopulateConvParameter() local
Ddedepthwise_conv2d_populate_v0.cc51 auto act_type = dedepthwise_conv2d_prim->activationType(); in PopulateDeconvDwParameter() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dfusion_eltwise.cc59 auto act_type = static_cast<ActivationType>( in CheckSupport() local
73 …auto act_type = static_cast<ActivationType>(reinterpret_cast<ActivationParameter *>(op_parameter)-… in CheckSupport() local
96 auto act_type = static_cast<ActivationType>( local
114 …auto act_type = static_cast<ActivationType>(reinterpret_cast<ActivationParameter *>(op_parameter)-… local
Dactivation.cc41 std::string ActivationOpenCLKernel::GetActTypeString(int act_type) { in GetActTypeString()
Dfusion_eltwise.h142 constexpr EltwiseOperator Activation2Operator(ActivationType act_type) { in Activation2Operator()
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dconv2d_populate.cc95 auto act_type = value->activation_type(); in PopulateConvParameter() local
Dadder_populate.cc68 auto act_type = value->activation_type(); in PopulateAdderParameter() local
Ddeconv2d_populate.cc108 auto act_type = value->activation_type(); in PopulateDeconvParameter() local
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dconvolution_base_npu.cc123 int ConvolutionBaseNPUOp::SetActivation(const ge::Operator *input, schema::ActivationType act_type)… in SetActivation()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/
Dnnacl_stream_utils.cc74 std::ostream &operator<<(std::ostream &code, ActType act_type) { in operator <<()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconvolution_winograd_fp32_coder.cc187 …nvolutionWinogradFP32Coder::GetOutputTransFunc(int input_unit, int output_unit, ActType act_type) { in GetOutputTransFunc()

12