Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceRNN.cpp110 T* stateOutput = context->getOutputBuffer<T>(kStateOutputTensor); in executeTyped()
154 NN_RET_CHECK(!context->isOmittedOutput(kStateOutputTensor)); in prepare()
159 NN_RET_CHECK(context->setOutputShape(kStateOutputTensor, outputStateShape)); in prepare()
/packages/modules/NeuralNetworks/common/types/operations/include/
DUnidirectionalSequenceRNN.h36 constexpr uint32_t kStateOutputTensor = 1; variable