Home
last modified time | relevance | path

Searched defs:auxInput (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp461 const float* auxInput = GetOptionalBuffer<const float>(aux_input_); in Eval() local
584 const _Float16* auxInput = GetOptionalBuffer<const _Float16>(aux_input_); in Eval() local
DBidirectionalSequenceRNN.cpp141 const T* auxInput = nullptr; in executeTyped() local
378 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare() local
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp2858 ANeuralNetworksOperandType auxInput = floatTensor3D; in lstmBidirectionalSequence() local