Home
last modified time | relevance | path

Searched refs:kOutputStateInTensor (Results 1 – 3 of 3) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp68 constexpr uint32_t kOutputStateInTensor = 18; variable
188 kOutputStateInTensor, in prepare()
320 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare()
384 Shape outputStateOutTensor = context->getInputShape(kOutputStateInTensor); in prepare()
401 const auto outputStateSize = getNumberOfElements(context->getInputShape(kOutputStateInTensor)); in execute()
452 context->getInputBuffer<float>(kOutputStateInTensor), in execute()
505 context->getInputBuffer<_Float16>(kOutputStateInTensor), in execute()
H A DLSTM.h88 static constexpr int kOutputStateInTensor = 18; variable
H A DLSTM.cpp83 output_state_in_ = GetInput(operation, operands, kOutputStateInTensor); in LSTMCell()
320 kOutputStateInTensor, in Prepare()