Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp213 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare() local
214 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
215 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells); in prepare()
216 const uint32_t outputSize = getSizeOfDimension(recurrentToOutputShape, 1); in prepare()
H A DQLSTM.cpp191 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare() local
192 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
193 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numUnits); in prepare()
194 const uint32_t outputSize = getSizeOfDimension(recurrentToOutputShape, 1); in prepare()