Searched refs:kProjClipParam (Results 1 – 8 of 8) sorted by relevance
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 65 constexpr uint32_t kProjClipParam = 22; variable
|
| D | BidirectionalSequenceLSTM.h | 131 static constexpr int kProjClipParam = 50; variable
|
| D | LSTM.h | 92 static constexpr int kProjClipParam = 22; variable
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| 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()
|
| D | UnidirectionalSequenceLSTM.cpp | 54 params.proj_clip = static_cast<float>(context->getInputValue<T>(kProjClipParam)); in getLSTMParams()
82 kProjClipParam, in prepare()
|
| 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()
|
| D | LayerNormLSTMTest.cpp | 256 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
|
| D | LSTMTest.cpp | 243 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
|