Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dl2_normalize_builder.cpp31 L2NormalizeBuilder::L2NormalizeBuilder() {} in L2NormalizeBuilder() function in OHOS::NeuralNetworkRuntime::Ops::L2NormalizeBuilder
33 L2NormalizeBuilder::~L2NormalizeBuilder() {} in ~L2NormalizeBuilder()
35 OH_NN_ReturnCode L2NormalizeBuilder::SetAxis(const std::shared_ptr<NNTensor>& tensor) in SetAxis()
61 OH_NN_ReturnCode L2NormalizeBuilder::SetEpsilon(const std::shared_ptr<NNTensor>& tensor) in SetEpsilon()
83 OH_NN_ReturnCode L2NormalizeBuilder::SetActivationType(const std::shared_ptr<NNTensor>& tensor) in SetActivationType()
111 OH_NN_ReturnCode L2NormalizeBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()
157 LiteGraphPrimitvePtr L2NormalizeBuilder::GetPrimitive() in GetPrimitive()
169 REGISTER_OPS(L2NormalizeBuilder, OH_NN_OPS_L2_NORMALIZE);
H A Dl2_normalize_builder.h25 class L2NormalizeBuilder : public OpsBuilder {
27 typedef OH_NN_ReturnCode (L2NormalizeBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
29 L2NormalizeBuilder();
30 ~L2NormalizeBuilder() override;
48 {OH_NN_L2_NORMALIZE_ACTIVATION_TYPE, &L2NormalizeBuilder::SetActivationType},
49 {OH_NN_L2_NORMALIZE_EPSILON, &L2NormalizeBuilder::SetEpsilon},
50 {OH_NN_L2_NORMALIZE_AXIS, &L2NormalizeBuilder::SetAxis}
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dl2_normalize_test.cpp39 L2NormalizeBuilder m_builder;