Home
last modified time | relevance | path

Searched refs:activationType (Results 1 - 23 of 23) sorted by relevance

/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dhswish_builder.cpp73 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_HSWISH; in GetPrimitive() local
78 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, in GetPrimitive()
H A Drelu6_builder.cpp77 mindspore::lite::ActivationType activationType{mindspore::lite::ACTIVATION_TYPE_RELU6}; in GetPrimitive()
79 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in GetPrimitive()
H A Dswish_builder.cpp74 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_SWISH}; in GetPrimitive()
76 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in GetPrimitive()
H A Dhard_sigmoid_builder.cpp73 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_HSIGMOID; in GetPrimitive() local
78 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, in GetPrimitive()
H A Dabs_builder.cpp71 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_ABS}; in GetPrimitive()
73 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in GetPrimitive()
H A Dsigmoid_builder.cpp78 mindspore::lite::ActivationType activationType{mindspore::lite::ACTIVATION_TYPE_SIGMOID}; in GetPrimitive()
80 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, minVal, in GetPrimitive()
H A Drelu_builder.cpp77 mindspore::lite::ActivationType activationType{mindspore::lite::ACTIVATION_TYPE_RELU}; in GetPrimitive()
79 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in GetPrimitive()
H A Dgelu_builder.cpp109 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_GELU; in GetPrimitive() local
113 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, in GetPrimitive()
H A Dleaky_relu_builder.cpp110 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_LEAKY_RELU}; in GetPrimitive()
112 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in GetPrimitive()
/foundation/ai/neural_network_runtime/example/drivers/nnrt/v2_0/hdi_cpu_service/src/
H A Dnode_functions.cpp42 attr->activation_type = static_cast<mindspore::schema::ActivationType>(addAttr.activationType); in GetAddPrimitive()
71 attr->activation_type = static_cast<mindspore::schema::ActivationType>(avgPoolAttr.activationType); in GetAvgPoolPrimitive()
124 attr->activation_type = static_cast<mindspore::schema::ActivationType>(conv2dAttr.activationType); in GetConv2dPrimitive()
151 attr->activation_type = static_cast<mindspore::schema::ActivationType>(fullConnAttr.activationType); in GetFullConnectionPrimitive()
181 attr->activation_type = static_cast<mindspore::schema::ActivationType>(maxPoolAttr.activationType); in GetMaxPoolFusionPrimitive()
207 attr->activation_type = static_cast<mindspore::schema::ActivationType>(matmulAttr.activationType); in GetMatMulFusionPrimitive()
277 attr->activation_type = static_cast<mindspore::schema::ActivationType>(scaleAttr.activationType); in GetScaleFusionPrimitive()
304 attr->activation_type = static_cast<mindspore::schema::ActivationType>(actAttr.activationType); in GetActivationPrimitive()
352 attr->activation_type = static_cast<mindspore::schema::ActivationType>(mulAttr.activationType); in GetMulFusionPrimitive()
/foundation/ai/neural_network_runtime/example/drivers/nnrt/v1_0/hdi_cpu_service/src/
H A Dnode_functions.cpp42 attr->activation_type = static_cast<mindspore::schema::ActivationType>(addAttr.activationType); in GetAddPrimitive()
71 attr->activation_type = static_cast<mindspore::schema::ActivationType>(avgPoolAttr.activationType); in GetAvgPoolPrimitive()
124 attr->activation_type = static_cast<mindspore::schema::ActivationType>(conv2dAttr.activationType); in GetConv2dPrimitive()
151 attr->activation_type = static_cast<mindspore::schema::ActivationType>(fullConnAttr.activationType); in GetFullConnectionPrimitive()
181 attr->activation_type = static_cast<mindspore::schema::ActivationType>(maxPoolAttr.activationType); in GetMaxPoolFusionPrimitive()
207 attr->activation_type = static_cast<mindspore::schema::ActivationType>(matmulAttr.activationType); in GetMatMulFusionPrimitive()
277 attr->activation_type = static_cast<mindspore::schema::ActivationType>(scaleAttr.activationType); in GetScaleFusionPrimitive()
304 attr->activation_type = static_cast<mindspore::schema::ActivationType>(actAttr.activationType); in GetActivationPrimitive()
352 attr->activation_type = static_cast<mindspore::schema::ActivationType>(mulAttr.activationType); in GetMulFusionPrimitive()
/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dinner_model_test.cpp1028 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_ABS}; in HWTEST_F()
1030 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in HWTEST_F()
1134 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
1139 padMode, roundMode, format, global, activationType); in HWTEST_F()
1356 mindspore::lite::ActivationType activationType{mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
1359 activationType, outputPaddings); in HWTEST_F()
1619 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
1621 axis, activationType); in HWTEST_F()
1675 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_GELU; in HWTEST_F() local
1680 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, in HWTEST_F()
1735 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_HSIGMOID; HWTEST_F() local
[all...]
/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/device_manager/
H A Ddevice_manager_test.cpp171 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_ABS}; in HWTEST_F()
173 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in HWTEST_F()
277 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
282 padMode, roundMode, format, global, activationType); in HWTEST_F()
499 mindspore::lite::ActivationType activationType{mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
502 activationType, outputPaddings); in HWTEST_F()
762 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
764 axis, activationType); in HWTEST_F()
818 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_GELU; in HWTEST_F() local
823 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, in HWTEST_F()
878 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_HSIGMOID; HWTEST_F() local
[all...]
/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dhard_sigmoid_test.cpp166 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_HSIGMOID; in HWTEST_F() local
168 EXPECT_EQ(returnValue, activationType); in HWTEST_F()
H A Dhswish_builder_test.cpp166 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_HSWISH; in HWTEST_F() local
168 EXPECT_EQ(returnValue, activationType); in HWTEST_F()
H A Drelu6_builder_test.cpp174 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_RELU6; in HWTEST_F() local
176 EXPECT_EQ(returnValue, activationType); in HWTEST_F()
H A Drelu_builder_test.cpp174 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_RELU; in HWTEST_F() local
176 EXPECT_EQ(returnValue, activationType); in HWTEST_F()
H A Dsigmoid_builder_test.cpp174 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_SIGMOID; in HWTEST_F() local
176 EXPECT_EQ(returnValue, activationType); in HWTEST_F()
H A Dgelu_builder_test.cpp244 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_GELU; in HWTEST_F() local
246 EXPECT_EQ(returnValue, activationType); in HWTEST_F()
/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dinner_model_test.cpp1089 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_ABS}; in HWTEST_F()
1091 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in HWTEST_F()
1195 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
1200 padMode, roundMode, format, global, activationType); in HWTEST_F()
1417 mindspore::lite::ActivationType activationType{mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
1420 activationType, outputPaddings); in HWTEST_F()
1680 mindspore::lite::ActivationType activationType {mindspore::lite::ACTIVATION_TYPE_NO_ACTIVATION}; in HWTEST_F()
1682 axis, activationType); in HWTEST_F()
1736 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_GELU; in HWTEST_F() local
1741 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, in HWTEST_F()
1796 mindspore::lite::ActivationType activationType = mindspore::lite::ACTIVATION_TYPE_HSIGMOID; HWTEST_F() local
[all...]
/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dlite_graph_to_hdi_model_v2_0.cpp40 activation.activationType = static_cast<HDI::Nnrt::V2_0::ActivationType>( in ConvertActivation()
62 addFusion.activationType = static_cast<HDI::Nnrt::V2_0::ActivationType>( in ConvertAddFusion()
107 avgPoolFusion.activationType = in ConvertAvgPoolFusion()
197 conv2DFusion.activationType = static_cast<ActivationType>( in ConvertConv2DFusion()
224 conv2dTransposeFusion.activationType = static_cast<ActivationType>( in ConvertConv2dTransposeFusion()
243 divFusion.activationType = static_cast<ActivationType>( in ConvertDivFusion()
309 fullConnection.activationType = static_cast<ActivationType>( in ConvertFullConnection()
395 matMulFusion.activationType = static_cast<ActivationType>( in ConvertMatMulFusion()
435 maxPoolFusion.activationType = static_cast<ActivationType>( in ConvertMaxPoolFusion()
453 mulFusion.activationType in ConvertMulFusion()
[all...]
H A Dlite_graph_to_hdi_model_v1_0.cpp40 activation.activationType = static_cast<HDI::Nnrt::V1_0::ActivationType>( in ConvertActivation()
62 addFusion.activationType = static_cast<HDI::Nnrt::V1_0::ActivationType>( in ConvertAddFusion()
107 avgPoolFusion.activationType = in ConvertAvgPoolFusion()
197 conv2DFusion.activationType = static_cast<ActivationType>( in ConvertConv2DFusion()
224 conv2dTransposeFusion.activationType = static_cast<ActivationType>( in ConvertConv2dTransposeFusion()
243 divFusion.activationType = static_cast<ActivationType>( in ConvertDivFusion()
309 fullConnection.activationType = static_cast<ActivationType>( in ConvertFullConnection()
395 matMulFusion.activationType = static_cast<ActivationType>( in ConvertMatMulFusion()
434 maxPoolFusion.activationType = static_cast<ActivationType>( in ConvertMaxPoolFusion()
452 mulFusion.activationType in ConvertMulFusion()
[all...]
H A Dlite_graph_to_hdi_model_v2_1.cpp40 activation.activationType = static_cast<HDI::Nnrt::V2_1::ActivationType>( in ConvertActivation()
62 addFusion.activationType = static_cast<HDI::Nnrt::V2_1::ActivationType>( in ConvertAddFusion()
141 avgPoolFusion.activationType = in ConvertAvgPoolFusion()
281 conv2DFusion.activationType = static_cast<ActivationType>( in ConvertConv2DFusion()
308 conv2dTransposeFusion.activationType = static_cast<ActivationType>( in ConvertConv2dTransposeFusion()
430 divFusion.activationType = static_cast<ActivationType>( in ConvertDivFusion()
563 fullConnection.activationType = static_cast<ActivationType>( in ConvertFullConnection()
837 l2NormalizeFusion.activationType = static_cast<ActivationType>( in ConvertL2NormalizeFusion()
857 matMulFusion.activationType = static_cast<ActivationType>( in ConvertMatMulFusion()
897 maxPoolFusion.activationType in ConvertMaxPoolFusion()
[all...]

Completed in 19 milliseconds