Searched refs:SetDropout (Results 1 – 2 of 2) sorted by relevance
46 OH_NN_ReturnCode SetDropout(const std::shared_ptr<NNTensor>& tensor);71 {OH_NN_LSTM_DROPOUT, &LSTMBuilder::SetDropout},
163 OH_NN_ReturnCode LSTMBuilder::SetDropout(const std::shared_ptr<NNTensor>& tensor) in SetDropout() function in OHOS::NeuralNetworkRuntime::Ops::LSTMBuilder