Searched refs:kInputGateBiasTensor (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 179 NN_RET_CHECK(hasTensor(context, kInputGateBiasTensor)); in prepare()
180 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
184 NN_RET_CHECK(!hasTensor(context, kInputGateBiasTensor)) in prepare()
337 context->getInputBuffer<float>(kInputGateBiasTensor), in execute()
390 context->getInputBuffer<_Float16>(kInputGateBiasTensor), in execute()
|
| D | QLSTM.cpp | 150 NN_RET_CHECK(hasTensor(context, kInputGateBiasTensor)); in prepare()
151 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare()
155 NN_RET_CHECK(!hasTensor(context, kInputGateBiasTensor)) in prepare()
303 reinterpret_cast<const int32_t*>(context->getInputBuffer(kInputGateBiasTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 237 inputGateBias_ = GetInput(operation, operands, kInputGateBiasTensor); in QuantizedLSTMCell()
296 auto inputGateBias = GetInput(operation, operands, kInputGateBiasTensor); in prepare()
|
| D | QuantizedLSTMTest.cpp | 124 setInputTensor(&execution, QuantizedLSTMCell::kInputGateBiasTensor, inputGateBias_), in invoke()
|
| D | LSTM.cpp | 75 input_gate_bias_ = GetInput(operation, operands, kInputGateBiasTensor); in LSTMCell()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 52 static constexpr int kInputGateBiasTensor = 9; variable
|
| D | UnidirectionalSequenceLSTM.h | 48 constexpr uint32_t kInputGateBiasTensor = 12; // Optional variable
|
| D | QLSTM.h | 47 constexpr uint32_t kInputGateBiasTensor = 12; variable
|
| D | LSTM.h | 77 static constexpr int kInputGateBiasTensor = 12; // Optional variable
|