Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp48 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
183 kRecurrentToCellWeightsTensor, in prepare()
245 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
435 context->getInputBuffer<float>(kRecurrentToCellWeightsTensor), in execute()
488 context->getInputBuffer<_Float16>(kRecurrentToCellWeightsTensor), in execute()
H A DQuantizedLSTM.h48 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
H A DQLSTM.cpp47 constexpr uint32_t kRecurrentToCellWeightsTensor = 7; variable
166 kRecurrentToCellWeightsTensor, in prepare()
223 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare()
381 const Shape recurrentToCellWeightsShape = context->getInputShape(kRecurrentToCellWeightsTensor); in execute()
435 reinterpret_cast<const int8_t*>(context->getInputBuffer(kRecurrentToCellWeightsTensor)); in execute()
H A DQuantizedLSTM.cpp230 recurrentToCellWeights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in QuantizedLSTMCell()
285 auto recurrentToCellWeights = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in prepare()
H A DLSTM.h69 static constexpr int kRecurrentToCellWeightsTensor = 7; variable
H A DQuantizedLSTMTest.cpp117 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToCellWeightsTensor, in invoke()
H A DLSTM.cpp66 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
315 kRecurrentToCellWeightsTensor, in Prepare()