Searched refs:ConcatBuilder (Results 1 – 4 of 4) sorted by relevance
27 ConcatBuilder::ConcatBuilder() {} in ConcatBuilder() function in OHOS::NeuralNetworkRuntime::Ops::ConcatBuilder29 ConcatBuilder::~ConcatBuilder() {} in ~ConcatBuilder()31 OH_NN_ReturnCode ConcatBuilder::SetAxis(const std::shared_ptr<NNTensor>& tensor) in SetAxis()55 OH_NN_ReturnCode ConcatBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()109 OH_NN_ReturnCode ConcatBuilder::SetInputsAndOutputs(const std::vector<uint32_t>& inputsIndex, in SetInputsAndOutputs()139 LiteGraphPrimitvePtr ConcatBuilder::GetPrimitive() in GetPrimitive()151 REGISTER_OPS(ConcatBuilder, OH_NN_OPS_CONCAT);
27 class ConcatBuilder : public OpsBuilder {29 typedef OH_NN_ReturnCode (ConcatBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);31 ConcatBuilder();32 ~ConcatBuilder() override;48 {OH_NN_CONCAT_AXIS, &ConcatBuilder::SetAxis}
36 ConcatBuilder m_builder;