Searched refs:LSTMBuilder (Results 1 – 3 of 3) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | lstm_builder.h | 27 class LSTMBuilder : public OpsBuilder {
31 LSTMBuilder();
32 ~LSTMBuilder() override;
65 {OH_NN_LSTM_BIDIRECTIONAL, &LSTMBuilder::SetBidirectional},
66 {OH_NN_LSTM_HAS_BIAS, &LSTMBuilder::SetHasBias},
67 {OH_NN_LSTM_INPUT_SIZE, &LSTMBuilder::SetInputSize},
68 {OH_NN_LSTM_HIDDEN_SIZE, &LSTMBuilder::SetHiddenSize},
69 {OH_NN_LSTM_NUM_LAYERS, &LSTMBuilder::SetNumLayers},
71 {OH_NN_LSTM_DROPOUT, &LSTMBuilder::SetDropout},
72 {OH_NN_LSTM_ZONEOUT_CELL, &LSTMBuilder::SetZoneoutCell},
[all …]
|
| H A D | lstm_builder.cpp | 27 LSTMBuilder::LSTMBuilder() {} in LSTMBuilder() function in OHOS::NeuralNetworkRuntime::Ops::LSTMBuilder
29 LSTMBuilder::~LSTMBuilder() {} in ~LSTMBuilder()
31 OH_NN_ReturnCode LSTMBuilder::SetBidirectional(const std::shared_ptr<NNTensor>& tensor) in SetBidirectional()
53 OH_NN_ReturnCode LSTMBuilder::SetHasBias(const std::shared_ptr<NNTensor>& tensor) in SetHasBias()
75 OH_NN_ReturnCode LSTMBuilder::SetInputSize(const std::shared_ptr<NNTensor>& tensor) in SetInputSize()
97 OH_NN_ReturnCode LSTMBuilder::SetHiddenSize(const std::shared_ptr<NNTensor>& tensor) in SetHiddenSize()
119 OH_NN_ReturnCode LSTMBuilder::SetNumLayers(const std::shared_ptr<NNTensor>& tensor) in SetNumLayers()
163 OH_NN_ReturnCode LSTMBuilder::SetDropout(const std::shared_ptr<NNTensor>& tensor) in SetDropout()
273 OH_NN_ReturnCode LSTMBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()
309 LiteGraphPrimitvePtr LSTMBuilder::GetPrimitive() in GetPrimitive()
[all …]
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | lstm_test.cpp | 57 LSTMBuilder m_builder;
|