Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dlstm_builder.h44 OH_NN_ReturnCode SetNumLayers(const std::shared_ptr<NNTensor>& tensor);
69 {OH_NN_LSTM_NUM_LAYERS, &LSTMBuilder::SetNumLayers},
H A Dlstm_builder.cpp119 OH_NN_ReturnCode LSTMBuilder::SetNumLayers(const std::shared_ptr<NNTensor>& tensor) in SetNumLayers() function in OHOS::NeuralNetworkRuntime::Ops::LSTMBuilder