Searched refs:kInputGateBiasTensor (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 55 constexpr uint32_t kInputGateBiasTensor = 12; // Optional variable
285 NN_RET_CHECK(hasTensor(context, kInputGateBiasTensor)); in prepare()
286 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
290 NN_RET_CHECK(!hasTensor(context, kInputGateBiasTensor)) in prepare()
443 context->getInputBuffer<float>(kInputGateBiasTensor), in execute()
496 context->getInputBuffer<_Float16>(kInputGateBiasTensor), in execute()
|
| D | QuantizedLSTM.h | 52 static constexpr int kInputGateBiasTensor = 9; variable
|
| D | QLSTM.cpp | 54 constexpr uint32_t kInputGateBiasTensor = 12; variable
265 NN_RET_CHECK(hasTensor(context, kInputGateBiasTensor)); in prepare()
266 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
270 NN_RET_CHECK(!hasTensor(context, kInputGateBiasTensor)) in prepare()
417 reinterpret_cast<const int32_t*>(context->getInputBuffer(kInputGateBiasTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 236 inputGateBias_ = GetInput(operation, operands, kInputGateBiasTensor); in QuantizedLSTMCell()
295 auto inputGateBias = GetInput(operation, operands, kInputGateBiasTensor); in prepare()
|
| D | LSTM.h | 78 static constexpr int kInputGateBiasTensor = 12; // Optional variable
|
| D | QuantizedLSTMTest.cpp | 123 setInputTensor(&execution, QuantizedLSTMCell::kInputGateBiasTensor, inputGateBias_), in invoke()
|
| D | LSTM.cpp | 75 input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor); in LSTMCell()
|