Searched refs:kRecurrentToForgetWeightsTensor (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 72 kRecurrentToForgetWeightsTensor, in prepare()
132 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
325 context->getInputBuffer<float>(kRecurrentToForgetWeightsTensor), in execute()
378 context->getInputBuffer<_Float16>(kRecurrentToForgetWeightsTensor), in execute()
|
| D | QLSTM.cpp | 49 kRecurrentToForgetWeightsTensor, in prepare()
103 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare()
261 context->getInputShape(kRecurrentToForgetWeightsTensor); in execute()
308 context->getInputBuffer(kRecurrentToForgetWeightsTensor)); in execute()
|
| D | QuantizedLSTM.cpp | 233 recurrentToForgetWeights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in QuantizedLSTMCell()
288 auto recurrentToForgetWeights = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in prepare()
|
| D | QuantizedLSTMTest.cpp | 114 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToForgetWeightsTensor, in invoke()
|
| D | LSTM.cpp | 65 recurrent_to_forget_weights_ = GetInput(operation, operands, kRecurrentToForgetWeightsTensor); in LSTMCell()
314 kRecurrentToForgetWeightsTensor, in Prepare()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 47 static constexpr int kRecurrentToForgetWeightsTensor = 6; variable
|
| D | UnidirectionalSequenceLSTM.h | 38 constexpr uint32_t kRecurrentToForgetWeightsTensor = 6; variable
|
| D | QLSTM.h | 36 constexpr uint32_t kRecurrentToForgetWeightsTensor = 6; variable
|
| D | LSTM.h | 67 static constexpr int kRecurrentToForgetWeightsTensor = 6; variable
|