Searched refs:StackBuilder (Results 1 – 3 of 3) sorted by relevance
28 StackBuilder::StackBuilder() {} in StackBuilder() function in OHOS::NeuralNetworkRuntime::Ops::StackBuilder30 StackBuilder::~StackBuilder() {} in ~StackBuilder()32 OH_NN_ReturnCode StackBuilder::SetAxis(const std::shared_ptr<NNTensor>& tensor) in SetAxis()60 OH_NN_ReturnCode StackBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()109 LiteGraphTensorPtr StackBuilder::GetPrimitive() in GetPrimitive()126 REGISTER_OPS(StackBuilder, OH_NN_OPS_STACK);
25 class StackBuilder : public OpsBuilder {27 typedef OH_NN_ReturnCode (StackBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);29 StackBuilder();30 ~StackBuilder() override;44 {OH_NN_STACK_AXIS, &StackBuilder::SetAxis}
37 StackBuilder m_builder;