Searched refs:LeakyReluBuilder (Results 1 – 3 of 3) sorted by relevance
27 LeakyReluBuilder::LeakyReluBuilder() {} in LeakyReluBuilder() function in OHOS::NeuralNetworkRuntime::Ops::LeakyReluBuilder29 LeakyReluBuilder::~LeakyReluBuilder() {} in ~LeakyReluBuilder()31 OH_NN_ReturnCode LeakyReluBuilder::SetNegativeSlope(const std::shared_ptr<NNTensor>& tensor) in SetNegativeSlope()53 OH_NN_ReturnCode LeakyReluBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()99 LiteGraphPrimitvePtr LeakyReluBuilder::GetPrimitive() in GetPrimitive()118 REGISTER_OPS(LeakyReluBuilder, OH_NN_OPS_LEAKY_RELU);
27 class LeakyReluBuilder : public OpsBuilder {29 typedef OH_NN_ReturnCode (LeakyReluBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);31 LeakyReluBuilder();32 ~LeakyReluBuilder() override;46 {OH_NN_LEAKY_RELU_NEGATIVE_SLOPE, &LeakyReluBuilder::SetNegativeSlope}
37 LeakyReluBuilder m_builder;