Home
last modified time | relevance | path

Searched refs:pFuseData (Results 1 – 5 of 5) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dfullconnection_builder.cpp128 int8_t* pFuseData = static_cast<int8_t*>(tensor->GetBuffer()); in SetFullConnectionActivation() local
129 …HOS::NeuralNetworkRuntime::Validation::ValidateFuseType(static_cast<OH_NN_FuseType>(*pFuseData))) { in SetFullConnectionActivation()
133 m_activationType = NNToMS::TransfromFusionType((OH_NN_FuseType)(*pFuseData)); in SetFullConnectionActivation()
H A Dpooling_builder.cpp236 int8_t* pFuseData = static_cast<int8_t*>(buffer); in SetActivation() local
237 …HOS::NeuralNetworkRuntime::Validation::ValidateFuseType(static_cast<OH_NN_FuseType>(*pFuseData))) { in SetActivation()
241 auto fuseType = (OH_NN_FuseType)(*pFuseData); in SetActivation()
H A Ddepthwise_conv2d_native_builder.cpp77 int8_t* pFuseData = static_cast<int8_t*>(buffer); in SetActivation() local
78 …HOS::NeuralNetworkRuntime::Validation::ValidateFuseType(static_cast<OH_NN_FuseType>(*pFuseData))) { in SetActivation()
82 m_activationType = NNToMS::TransfromFusionType((OH_NN_FuseType)(*pFuseData)); in SetActivation()
H A Dconv2d_builder.cpp216 int8_t* pFuseData = static_cast<int8_t*>(buffer); in SetActavitation() local
217 …HOS::NeuralNetworkRuntime::Validation::ValidateFuseType(static_cast<OH_NN_FuseType>(*pFuseData))) { in SetActavitation()
221 m_activationType = NNToMS::TransfromFusionType((OH_NN_FuseType)(*pFuseData)); in SetActavitation()
H A Dconv2d_transpose_builder.cpp230 int8_t* pFuseData = static_cast<int8_t*>(buffer); in SetActivation() local
231 …HOS::NeuralNetworkRuntime::Validation::ValidateFuseType(static_cast<OH_NN_FuseType>(*pFuseData))) { in SetActivation()
235 m_activationType = NNToMS::TransfromFusionType((OH_NN_FuseType)(*pFuseData)); in SetActivation()