Home
last modified time | relevance | path

Searched refs:SetOutputNum (Results 1 – 2 of 2) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dsplit_builder.h45 OH_NN_ReturnCode SetOutputNum(const std::shared_ptr<NNTensor>& tensor);
54 {OH_NN_SPLIT_OUTPUT_NUM, &SplitBuilder::SetOutputNum},
H A Dsplit_builder.cpp86 OH_NN_ReturnCode SplitBuilder::SetOutputNum(const std::shared_ptr<NNTensor>& tensor) in SetOutputNum() function in OHOS::NeuralNetworkRuntime::Ops::SplitBuilder