Searched refs:kCellClipParam (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 71 constexpr uint32_t kCellClipParam = 21; variable
106 params.cell_clip = static_cast<float>(context->getInputValue<T>(kCellClipParam)); in getLSTMParams()
188 kCellClipParam, in prepare()
|
| D | BidirectionalSequenceLSTM.h | 133 static constexpr int kCellClipParam = 49; variable
|
| D | LSTM.h | 92 static constexpr int kCellClipParam = 21; variable
|
| D | BidirectionalSequenceLSTM.cpp | 175 const auto& clipOperand = *GetInput(operation, operands, kCellClipParam); in BidirectionalSequenceLSTM()
239 kCellClipParam, in Prepare()
252 const auto& cellOperand = *GetInput(operation, operands, 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 | 253 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
|
| D | LSTMTest.cpp | 242 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
|