Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp59 constexpr uint32_t kBwOutputHiddenStateTensor = 3; variable
248 bwOutputHiddenState = context->getOutputBuffer<T>(kBwOutputHiddenStateTensor - delta); in executeTyped()
459 NN_RET_CHECK(context->setOutputShape(kBwOutputHiddenStateTensor - delta, in prepare()