Searched refs:SplitBuilder (Results 1 – 3 of 3) sorted by relevance
25 SplitBuilder::SplitBuilder() {} in SplitBuilder() function in OHOS::NeuralNetworkRuntime::Ops::SplitBuilder27 SplitBuilder::~SplitBuilder() {} in ~SplitBuilder()29 OH_NN_ReturnCode SplitBuilder::SetInputAndOutput(const std::vector<uint32_t> &inputsIndex, in SetInputAndOutput()64 OH_NN_ReturnCode SplitBuilder::SetAxis(const std::shared_ptr<NNTensor>& tensor) in SetAxis()86 OH_NN_ReturnCode SplitBuilder::SetOutputNum(const std::shared_ptr<NNTensor>& tensor) in SetOutputNum()103 OH_NN_ReturnCode SplitBuilder::SetSizeSplits(const std::shared_ptr<NNTensor>& tensor) in SetSizeSplits()124 OH_NN_ReturnCode SplitBuilder::Build(const std::vector<uint32_t> ¶msIndex, in Build()167 LiteGraphTensorPtr SplitBuilder::GetPrimitive() in GetPrimitive()184 REGISTER_OPS(SplitBuilder, OH_NN_OPS_SPLIT);
27 class SplitBuilder : public OpsBuilder {29 typedef OH_NN_ReturnCode (SplitBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);31 SplitBuilder();32 ~SplitBuilder() override;53 {OH_NN_SPLIT_AXIS, &SplitBuilder::SetAxis},54 {OH_NN_SPLIT_OUTPUT_NUM, &SplitBuilder::SetOutputNum},55 {OH_NN_SPLIT_SIZE_SPLITS, &SplitBuilder::SetSizeSplits}
41 SplitBuilder m_builder;