Searched refs:kInputToForgetWeightsTensor (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 39 constexpr uint32_t kInputToForgetWeightsTensor = 2; variable
176 kInputToForgetWeightsTensor, in prepare()
222 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
426 context->getInputBuffer<float>(kInputToForgetWeightsTensor), in execute()
479 context->getInputBuffer<_Float16>(kInputToForgetWeightsTensor), in execute()
|
| D | QuantizedLSTM.h | 41 static constexpr int kInputToForgetWeightsTensor = 2; variable
|
| D | QLSTM.cpp | 37 constexpr uint32_t kInputToForgetWeightsTensor = 2; variable
161 kInputToForgetWeightsTensor, in prepare()
202 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
373 const Shape inputToForgetWeightsShape = context->getInputShape(kInputToForgetWeightsTensor); in execute()
420 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToForgetWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 227 inputToForgetWeights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in QuantizedLSTMCell()
278 auto inputToForgetWeights = GetInput(operation, operands, kInputToForgetWeightsTensor); in prepare()
|
| D | LSTM.h | 62 static constexpr int kInputToForgetWeightsTensor = 2; variable
|
| D | QuantizedLSTMTest.cpp | 101 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToForgetWeightsTensor, in invoke()
|
| D | LSTM.cpp | 59 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
311 kInputToForgetWeightsTensor, in Prepare()
|