Searched refs:kOutputGateBiasTensor (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 56 constexpr uint32_t kOutputGateBiasTensor = 15; variable
171 kOutputGateBiasTensor, in prepare()
287 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare()
403 context->getInputBuffer<float>(kOutputGateBiasTensor), in execute()
445 context->getInputBuffer<_Float16>(kOutputGateBiasTensor), in execute()
|
| D | QuantizedLSTM.h | 42 static constexpr int kOutputGateBiasTensor = 12; variable
|
| D | QuantizedLSTM.cpp | 233 outputGateBias_ = GetInput(operation, operands, kOutputGateBiasTensor); in QuantizedLSTMCell()
306 auto outputGateBias = GetInput(operation, operands, kOutputGateBiasTensor); in prepare()
|
| D | LSTM.h | 79 static constexpr int kOutputGateBiasTensor = 15; variable
|
| D | QuantizedLSTMTest.cpp | 130 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kOutputGateBiasTensor, in invoke()
|
| D | LSTM.cpp | 73 output_gate_bias_ = GetInput(operation, operands, kOutputGateBiasTensor); in LSTMCell()
|