Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp44 constexpr uint32_t kAuxInputTensor = 9; // optional variable
98 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode()
147 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
151 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
156 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
378 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
DBidirectionalSequenceLSTM.h120 static constexpr int kAuxInputTensor = 39; // Optional variable
DBidirectionalSequenceLSTM.cpp149 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()