Searched refs:kProjClipParam (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 72 constexpr uint32_t kProjClipParam = 22; variable
107 params.proj_clip = static_cast<float>(context->getInputValue<T>(kProjClipParam)); in getLSTMParams()
189 kProjClipParam, in prepare()
|
| D | BidirectionalSequenceLSTM.h | 134 static constexpr int kProjClipParam = 50; variable
|
| D | LSTM.h | 93 static constexpr int kProjClipParam = 22; variable
|
| D | BidirectionalSequenceLSTM.cpp | 176 const auto& projOperand = *GetInput(operation, operands, kProjClipParam); in BidirectionalSequenceLSTM()
240 kProjClipParam, in Prepare()
253 const auto& projOperand = *GetInput(operation, operands, 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 | 255 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
|
| D | LSTMTest.cpp | 244 ASSERT_EQ(execution.setInput(LSTMCell::kProjClipParam, &proj_clip_, sizeof(proj_clip_)), in Invoke()
|