Home
last modified time | relevance | path

Searched refs:kInputGateBiasTensor (Results 1 – 7 of 7) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp57 constexpr uint32_t kInputGateBiasTensor = 12; // Optional variable
288 NN_RET_CHECK(hasTensor(context, kInputGateBiasTensor)); in prepare()
289 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
293 NN_RET_CHECK(!hasTensor(context, kInputGateBiasTensor)) in prepare()
446 context->getInputBuffer<float>(kInputGateBiasTensor), in execute()
499 context->getInputBuffer<_Float16>(kInputGateBiasTensor), in execute()
H A DQuantizedLSTM.h52 static constexpr int kInputGateBiasTensor = 9; variable
H A DQLSTM.cpp57 constexpr uint32_t kInputGateBiasTensor = 12; variable
266 NN_RET_CHECK(hasTensor(context, kInputGateBiasTensor)); in prepare()
267 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
271 NN_RET_CHECK(!hasTensor(context, kInputGateBiasTensor)) in prepare()
419 reinterpret_cast<const int32_t*>(context->getInputBuffer(kInputGateBiasTensor)); in execute()
H A DQuantizedLSTM.cpp233 inputGateBias_ = GetInput(operation, operands, kInputGateBiasTensor); in QuantizedLSTMCell()
292 auto inputGateBias = GetInput(operation, operands, kInputGateBiasTensor); in prepare()
H A DLSTM.h78 static constexpr int kInputGateBiasTensor = 12; // Optional variable
H A DQuantizedLSTMTest.cpp124 setInputTensor(&execution, QuantizedLSTMCell::kInputGateBiasTensor, inputGateBias_), in invoke()
H A DLSTM.cpp75 input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor); in LSTMCell()