Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceRNN.h53 constexpr uint32_t kBwOutputHiddenStateTensor = 3; variable
/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp219 bwOutputHiddenState = context->getOutputBuffer<T>(kBwOutputHiddenStateTensor - delta); in executeTyped()
400 NN_RET_CHECK(context->setOutputShape(kBwOutputHiddenStateTensor - delta, in prepare()