Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp43 constexpr uint32_t kAuxInputTensor = 9; // optional variable
96 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode()
145 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
149 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
154 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
377 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
H A DBidirectionalSequenceLSTM.h117 static constexpr int kAuxInputTensor = 39; // Optional variable
H A DBidirectionalSequenceLSTM.cpp148 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()