Home
last modified time | relevance | path

Searched refs:SetApproximate (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dgelu_builder_test.cpp33 void SetApproximate(OH_NN_DataType dataType,
50 void GeluBuilderTest::SetApproximate(OH_NN_DataType dataType, in SetApproximate() function in OHOS::NeuralNetworkRuntime::UnitTest::GeluBuilderTest
69 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
84 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
104 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
122 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
167 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
203 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
236 SetApproximate(OH_NN_BOOL, m_paramsDim, nullptr, OH_NN_GELU_APPROXIMATE);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dgelu_builder.h37 OH_NN_ReturnCode SetApproximate(const std::shared_ptr<NNTensor>& tensor);
42 {OH_NN_GELU_APPROXIMATE, &GeluBuilder::SetApproximate}
H A Dgelu_builder.cpp34 OH_NN_ReturnCode GeluBuilder::SetApproximate(const std::shared_ptr<NNTensor>& tensor) in SetApproximate() function in OHOS::NeuralNetworkRuntime::Ops::GeluBuilder