Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp49 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
184 kRecurrentToOutputWeightsTensor, in prepare()
213 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
436 context->getInputBuffer<float>(kRecurrentToOutputWeightsTensor), in execute()
437 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
489 context->getInputBuffer<_Float16>(kRecurrentToOutputWeightsTensor), in execute()
490 context->getInputShape(kRecurrentToOutputWeightsTensor), in execute()
H A DQuantizedLSTM.h49 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
H A DQLSTM.cpp48 constexpr uint32_t kRecurrentToOutputWeightsTensor = 8; variable
167 kRecurrentToOutputWeightsTensor, in prepare()
191 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare()
385 context->getInputShape(kRecurrentToOutputWeightsTensor); in execute()
444 context->getInputBuffer(kRecurrentToOutputWeightsTensor)); in execute()
H A DQuantizedLSTM.cpp231 recurrentToOutputWeights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in QuantizedLSTMCell()
286 auto recurrentToOutputWeights = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in prepare()
H A DLSTM.h70 static constexpr int kRecurrentToOutputWeightsTensor = 8; variable
H A DQuantizedLSTMTest.cpp120 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kRecurrentToOutputWeightsTensor, in invoke()
H A DLSTM.cpp67 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
316 kRecurrentToOutputWeightsTensor, in Prepare()