Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp39 constexpr uint32_t kAuxInputTensor = 9; // optional variable
99 const bool hasAuxInputs = !context->isOmittedInput(kAuxInputTensor); in executeTyped()
101 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
105 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
301 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
305 const bool auxInputsAllOrNone = (context->isOmittedInput(kAuxInputTensor) && in prepare()
308 (!context->isOmittedInput(kAuxInputTensor) && in prepare()
312 const bool hasAuxInputs = !context->isOmittedInput(kAuxInputTensor); in prepare()
DBidirectionalSequenceLSTM.h118 static constexpr int kAuxInputTensor = 39; // Optional variable
DBidirectionalSequenceLSTM.cpp116 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()