Home
last modified time | relevance | path

Searched refs:fuseType (Results 1 – 10 of 10) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dnn_validation_test.cpp147 OH_NN_FuseType fuseType = (OH_NN_FuseType)fuseTypeTest; variable
148 EXPECT_EQ(true, ValidateFuseType(fuseType));
159 OH_NN_FuseType fuseType = (OH_NN_FuseType)fuseTypeTest; variable
160 EXPECT_EQ(false, ValidateFuseType(fuseType));
171 OH_NN_FuseType fuseType = (OH_NN_FuseType)fuseTypeTest; variable
172 EXPECT_EQ(false, ValidateFuseType(fuseType));
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dnn_validation_test.cpp159 OH_NN_FuseType fuseType = (OH_NN_FuseType)fuseTypeTest; variable
160 EXPECT_EQ(true, ValidateFuseType(fuseType));
171 OH_NN_FuseType fuseType = (OH_NN_FuseType)fuseTypeTest; variable
172 EXPECT_EQ(false, ValidateFuseType(fuseType));
183 OH_NN_FuseType fuseType = (OH_NN_FuseType)fuseTypeTest; variable
184 EXPECT_EQ(false, ValidateFuseType(fuseType));
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_core/
H A Dvalidation.cpp53 bool ValidateFuseType(OH_NN_FuseType fuseType) in ValidateFuseType() argument
55 if ((fuseType >= OH_NN_FUSED_NONE) && (fuseType <= OH_NN_FUSED_RELU6)) { in ValidateFuseType()
H A Dvalidation.h41 bool ValidateFuseType(OH_NN_FuseType fuseType);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Ddiv_builder.cpp58 auto fuseType = (OH_NN_FuseType)(*fuseData); in SetActicationType() local
59 m_activationType = NNToMS::TransfromFusionType(fuseType); in SetActicationType()
H A Dmul_builder.cpp60 auto fuseType = (OH_NN_FuseType)(*fuseData); in SetActivationType() local
61 m_activationType = NNToMS::TransfromFusionType(fuseType); in SetActivationType()
H A Dsub_builder.cpp56 auto fuseType = (OH_NN_FuseType)(*fuseData); in SetActivationType() local
57 m_activationType = NNToMS::TransfromFusionType(fuseType); in SetActivationType()
H A Dscale_builder.cpp83 auto fuseType = (OH_NN_FuseType)(*fuseData); in SetActivationType() local
84 m_activationType = NNToMS::TransfromFusionType(fuseType); in SetActivationType()
H A Dmatmul_builder.cpp106 auto fuseType = (OH_NN_FuseType)(*fuseData); in SetActivationType() local
107 m_activationType = NNToMS::TransfromFusionType(fuseType); in SetActivationType()
H A Dpooling_builder.cpp241 auto fuseType = (OH_NN_FuseType)(*pFuseData); in SetActivation() local
242 m_activationType = NNToMS::TransfromFusionType(fuseType); in SetActivation()