Searched refs:kProjClipParam (Results 1 – 7 of 7) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceLSTM.cpp | 74 constexpr uint32_t kProjClipParam = 22; variable
108 params.proj_clip = static_cast<float>(context->getInputValue<T>(kProjClipParam)); in getLSTMParams()
192 kProjClipParam, in prepare()
|
| H A D | BidirectionalSequenceLSTM.h | 131 static constexpr int kProjClipParam = 50; variable
|
| H A D | BidirectionalSequenceLSTM.cpp | 175 const auto& projOperand = *GetInput(operation, operands, kProjClipParam); in BidirectionalSequenceLSTM()
239 kProjClipParam, in Prepare()
252 const auto& projOperand = *GetInput(operation, operands, kProjClipParam); in Prepare()
|
| H A D | LSTM.h | 93 static constexpr int kProjClipParam = 22; variable
|
| H A D | LSTM.cpp | 91 const auto& projClipOperand = *GetInput(operation, operands, kProjClipParam); in LSTMCell()
324 kProjClipParam, in Prepare()
336 const auto& projClipOperand = *GetInput(operation, operands, kProjClipParam); in Prepare()
|
| H A D | LayerNormLSTMTest.cpp | 255 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
|
| H A D | LSTMTest.cpp | 243 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
|