Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp58 constexpr uint32_t kFwOutputHiddenStateTensor = 2; variable
247 fwOutputHiddenState = context->getOutputBuffer<T>(kFwOutputHiddenStateTensor - delta); in executeTyped()
457 NN_RET_CHECK(context->setOutputShape(kFwOutputHiddenStateTensor - delta, in prepare()