Searched refs:kOutputGateBiasTensor (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 77 kOutputGateBiasTensor, in prepare()
194 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
340 context->getInputBuffer<float>(kOutputGateBiasTensor), in execute()
393 context->getInputBuffer<_Float16>(kOutputGateBiasTensor), in execute()
|
| D | QuantizedLSTM.cpp | 240 outputGateBias_ = GetInput(operation, operands, kOutputGateBiasTensor); in QuantizedLSTMCell()
312 auto outputGateBias = GetInput(operation, operands, kOutputGateBiasTensor); in prepare()
|
| D | QLSTM.cpp | 54 kOutputGateBiasTensor, in prepare()
165 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
334 reinterpret_cast<const int32_t*>(context->getInputBuffer(kOutputGateBiasTensor)); in execute()
|
| D | QuantizedLSTMTest.cpp | 131 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kOutputGateBiasTensor, in invoke()
|
| D | LSTM.cpp | 78 output_gate_bias_ = GetInput(operation, operands, kOutputGateBiasTensor); in LSTMCell()
319 kOutputGateBiasTensor, in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 55 static constexpr int kOutputGateBiasTensor = 12; variable
|
| D | UnidirectionalSequenceLSTM.h | 51 constexpr uint32_t kOutputGateBiasTensor = 15; variable
|
| D | QLSTM.h | 50 constexpr uint32_t kOutputGateBiasTensor = 15; variable
|
| D | LSTM.h | 80 static constexpr int kOutputGateBiasTensor = 15; variable
|