Home
last modified time | relevance | path

Searched refs:kInputToForgetWeightsTensor (Results 1 – 7 of 7) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp41 constexpr uint32_t kInputToForgetWeightsTensor = 2; variable
179 kInputToForgetWeightsTensor, in prepare()
225 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
429 context->getInputBuffer<float>(kInputToForgetWeightsTensor), in execute()
482 context->getInputBuffer<_Float16>(kInputToForgetWeightsTensor), in execute()
H A DQuantizedLSTM.h41 static constexpr int kInputToForgetWeightsTensor = 2; variable
H A DQLSTM.cpp40 constexpr uint32_t kInputToForgetWeightsTensor = 2; variable
162 kInputToForgetWeightsTensor, in prepare()
203 const Shape inputToForgetShape = context->getInputShape(kInputToForgetWeightsTensor); in prepare()
375 const Shape inputToForgetWeightsShape = context->getInputShape(kInputToForgetWeightsTensor); in execute()
422 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToForgetWeightsTensor)); in execute()
H A DQuantizedLSTM.cpp224 inputToForgetWeights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in QuantizedLSTMCell()
275 auto inputToForgetWeights = GetInput(operation, operands, kInputToForgetWeightsTensor); in prepare()
H A DLSTM.h62 static constexpr int kInputToForgetWeightsTensor = 2; variable
H A DQuantizedLSTMTest.cpp102 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToForgetWeightsTensor, in invoke()
H A DLSTM.cpp59 input_to_forget_weights_ = GetInput(operation, operands, kInputToForgetWeightsTensor); in LSTMCell()
311 kInputToForgetWeightsTensor, in Prepare()