Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceRNN.h52 constexpr uint32_t kFwOutputHiddenStateTensor = 2; variable
/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp218 fwOutputHiddenState = context->getOutputBuffer<T>(kFwOutputHiddenStateTensor - delta); in executeTyped()
398 NN_RET_CHECK(context->setOutputShape(kFwOutputHiddenStateTensor - delta, in prepare()