Searched refs:kForgetGateBiasTensor (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 56 constexpr uint32_t kForgetGateBiasTensor = 13; variable
182 kForgetGateBiasTensor, in prepare()
294 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
444 context->getInputBuffer<float>(kForgetGateBiasTensor), in execute()
497 context->getInputBuffer<_Float16>(kForgetGateBiasTensor), in execute()
|
| D | QuantizedLSTM.h | 53 static constexpr int kForgetGateBiasTensor = 10; variable
|
| D | QLSTM.cpp | 55 constexpr uint32_t kForgetGateBiasTensor = 13; variable
167 kForgetGateBiasTensor, in prepare()
274 const Shape forgetGateBiasShape = context->getInputShape(kForgetGateBiasTensor); in prepare()
428 reinterpret_cast<const int32_t*>(context->getInputBuffer(kForgetGateBiasTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 237 forgetGateBias_ = GetInput(operation, operands, kForgetGateBiasTensor); in QuantizedLSTMCell()
309 auto forgetGateBias = GetInput(operation, operands, kForgetGateBiasTensor); in prepare()
|
| D | LSTM.h | 79 static constexpr int kForgetGateBiasTensor = 13; variable
|
| D | QuantizedLSTMTest.cpp | 125 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kForgetGateBiasTensor, in invoke()
|
| D | LSTM.cpp | 76 forget_gate_bias_ = GetInput(operation, operands, kForgetGateBiasTensor); in LSTMCell()
317 kForgetGateBiasTensor, in Prepare()
|