Searched refs:kCellClipParam (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 69 constexpr uint32_t kCellClipParam = 21; variable
99 params.cell_clip = static_cast<float>(context->getInputValue<T>(kCellClipParam)); in getLSTMParams()
175 kCellClipParam, in prepare()
|
| D | BidirectionalSequenceLSTM.h | 131 static constexpr int kCellClipParam = 49; variable
|
| D | LSTM.h | 90 static constexpr int kCellClipParam = 21; variable
|
| D | BidirectionalSequenceLSTM.cpp | 142 params_.cell_clip = getScalarData<float>(*GetInput(operation, operands, kCellClipParam)); in BidirectionalSequenceLSTM()
146 getScalarData<_Float16>(*GetInput(operation, operands, kCellClipParam))); in BidirectionalSequenceLSTM()
|
| D | LayerNormLSTMTest.cpp | 254 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, &cell_clip_, sizeof(cell_clip_)), in Invoke()
|
| D | LSTMTest.cpp | 239 ASSERT_EQ(execution.setInput(LSTMCell::kCellClipParam, in Invoke()
|
| D | LSTM.cpp | 84 params_.cell_clip = getScalarData<float>(*GetInput(operation, operands, kCellClipParam)); in LSTMCell()
88 getScalarData<_Float16>(*GetInput(operation, operands, kCellClipParam))); in LSTMCell()
|