Searched refs:FullConnectionBuilder (Results 1 – 4 of 4) sorted by relevance
29 FullConnectionBuilder::FullConnectionBuilder() {} in FullConnectionBuilder() function in OHOS::NeuralNetworkRuntime::Ops::FullConnectionBuilder31 FullConnectionBuilder::~FullConnectionBuilder() {} in ~FullConnectionBuilder()33 OH_NN_ReturnCode FullConnectionBuilder::SetFullConnectionInput(const std::vector<uint32_t>& inputsI… in SetFullConnectionInput()57 OH_NN_ReturnCode FullConnectionBuilder::SetHasBias(const std::shared_ptr<NNTensor>& tensor) in SetHasBias()79 OH_NN_ReturnCode FullConnectionBuilder::SetUseAxis(const std::shared_ptr<NNTensor>& tensor) in SetUseAxis()108 OH_NN_ReturnCode FullConnectionBuilder::SetFullConnectionActivation(const std::shared_ptr<NNTensor>… in SetFullConnectionActivation()138 OH_NN_ReturnCode FullConnectionBuilder::SetAxis(const std::shared_ptr<NNTensor>& tensor) in SetAxis()170 OH_NN_ReturnCode FullConnectionBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()215 LiteGraphPrimitvePtr FullConnectionBuilder::GetPrimitive() in GetPrimitive()228 REGISTER_OPS(FullConnectionBuilder, OH_NN_OPS_FULL_CONNECTION);
27 class FullConnectionBuilder : public OpsBuilder {29 typedef OH_NN_ReturnCode (FullConnectionBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);31 FullConnectionBuilder();32 ~FullConnectionBuilder() override;55 {OH_NN_FULL_CONNECTION_ACTIVATIONTYPE, &FullConnectionBuilder::SetFullConnectionActivation},56 {OH_NN_FULL_CONNECTION_HAS_BIAS, &FullConnectionBuilder::SetHasBias},57 {OH_NN_FULL_CONNECTION_USE_AXIS, &FullConnectionBuilder::SetUseAxis},58 {OH_NN_FULL_CONNECTION_AXIS, &FullConnectionBuilder::SetAxis}
40 FullConnectionBuilder m_builder;
43 FullConnectionBuilder m_builder;