Home
last modified time | relevance | path

Searched refs:kFwOutputTensor (Results 1 – 4 of 4) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp56 constexpr uint32_t kFwOutputTensor = 0; variable
162 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped()
163 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped()
306 context->getOutputBuffer<T>(kFwOutputTensor)); in executeTyped()
438 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare()
443 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
H A DBidirectionalSequenceLSTM.h147 static constexpr int kFwOutputTensor = 0; variable
H A DBidirectionalSequenceLSTM.cpp191 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
/aosp12/packages/modules/NeuralNetworks/common/
H A DCpuExecutor.cpp1002 operands[outs[BidirectionalSequenceLSTM::kFwOutputTensor]]; in executeOperation()