Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp43 constexpr uint32_t kInputToOutputWeightsTensor = 4; variable
181 kInputToOutputWeightsTensor, in prepare()
208 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
431 context->getInputBuffer<float>(kInputToOutputWeightsTensor), in execute()
432 context->getInputShape(kInputToOutputWeightsTensor), in execute()
484 context->getInputBuffer<_Float16>(kInputToOutputWeightsTensor), in execute()
485 context->getInputShape(kInputToOutputWeightsTensor), in execute()
H A DQuantizedLSTM.h43 static constexpr int kInputToOutputWeightsTensor = 4; variable
H A DQLSTM.cpp42 constexpr uint32_t kInputToOutputWeightsTensor = 4; variable
164 kInputToOutputWeightsTensor, in prepare()
186 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare()
383 const Shape inputToOutputWeightsShape = context->getInputShape(kInputToOutputWeightsTensor); in execute()
442 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToOutputWeightsTensor)); in execute()
H A DQuantizedLSTM.cpp226 inputToOutputWeights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in QuantizedLSTMCell()
277 auto inputToOutputWeights = GetInput(operation, operands, kInputToOutputWeightsTensor); in prepare()
H A DLSTM.h64 static constexpr int kInputToOutputWeightsTensor = 4; variable
H A DQuantizedLSTMTest.cpp108 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kInputToOutputWeightsTensor, in invoke()
H A DLSTM.cpp61 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
313 kInputToOutputWeightsTensor, in Prepare()