Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp57 constexpr uint32_t kBwOutputTensor = 1; // Only if mergeOutputs parameter is false variable
167 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()
168 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()
309 context->getOutputBuffer<T>(kBwOutputTensor)); in executeTyped()
445 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare()
450 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
H A DBidirectionalSequenceLSTM.h148 static constexpr int kBwOutputTensor = 1; // Ignored if merge_outputs is set. variable
H A DBidirectionalSequenceLSTM.cpp193 bw_output_ = GetOutput(operation, operands, kBwOutputTensor); in BidirectionalSequenceLSTM()
/aosp12/packages/modules/NeuralNetworks/common/
H A DCpuExecutor.cpp1013 operands[outs[BidirectionalSequenceLSTM::kBwOutputTensor]]; in executeOperation()