Home
last modified time | relevance | path

Searched refs:kRecurrentToInputWeightsTensor (Results 1 – 9 of 9) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp46 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; // Optional variable
234 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
235 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
253 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
255 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
433 context->getInputBuffer<float>(kRecurrentToInputWeightsTensor), in execute()
486 context->getInputBuffer<_Float16>(kRecurrentToInputWeightsTensor), in execute()
H A DQLSTM.cpp45 constexpr uint32_t kRecurrentToInputWeightsTensor = 5; variable
212 if (hasTensor(context, kRecurrentToInputWeightsTensor)) { in prepare()
213 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare()
231 hasTensor(context, kRecurrentToInputWeightsTensor)) || in prepare()
233 !hasTensor(context, kRecurrentToInputWeightsTensor)); in prepare()
372 context->getInputShape(kRecurrentToInputWeightsTensor); in execute()
413 context->getInputBuffer(kRecurrentToInputWeightsTensor)); in execute()
H A DQuantizedLSTM.h46 static constexpr int kRecurrentToInputWeightsTensor = 5; variable
H A DQuantizedLSTM.cpp228 recurrentToInputWeights_ = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in QuantizedLSTMCell()
283 auto recurrentToInputWeights = GetInput(operation, operands, kRecurrentToInputWeightsTensor); in prepare()
H A DLSTM.h67 static constexpr int kRecurrentToInputWeightsTensor = 5; // Optional variable
H A DLayerNormLSTMTest.cpp229 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
H A DQuantizedLSTMTest.cpp111 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToInputWeightsTensor, in invoke()
H A DLSTMTest.cpp217 execution.setInput(LSTMCell::kRecurrentToInputWeightsTensor, nullptr, 0); in Invoke()
H A DLSTM.cpp64 GetInput(operation, operands, kRecurrentToInputWeightsTensor); // optional in LSTMCell()