Searched refs:kCellClipParam (Results 1 – 8 of 8) sorted by relevance
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceLSTM.h | 64 constexpr uint32_t kCellClipParam = 21; variable
|
| D | BidirectionalSequenceLSTM.h | 130 static constexpr int kCellClipParam = 49; variable
|
| D | LSTM.h | 91 static constexpr int kCellClipParam = 21; variable
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceLSTM.cpp | 174 const auto& clipOperand = *GetInput(operation, operands, kCellClipParam); in BidirectionalSequenceLSTM()
238 kCellClipParam, in Prepare()
251 const auto& cellOperand = *GetInput(operation, operands, kCellClipParam); in Prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 53 params.cell_clip = static_cast<float>(context->getInputValue<T>(kCellClipParam)); in getLSTMParams()
81 kCellClipParam, in prepare()
|
| D | LSTM.cpp | 90 const auto& cellClipOperand = *GetInput(operation, operands, kCellClipParam); in LSTMCell()
323 kCellClipParam, in Prepare()
335 const auto& cellClipOperand = *GetInput(operation, operands, kCellClipParam); in Prepare()
|
| D | LayerNormLSTMTest.cpp | 254 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
|
| D | LSTMTest.cpp | 241 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
|