Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp289 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare() local
290 NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1); in prepare()
291 NN_RET_CHECK_EQ(getSizeOfDimension(inputGateBiasShape, 0), numCells); in prepare()
H A DQLSTM.cpp267 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare() local
268 NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1); in prepare()
269 NN_RET_CHECK_EQ(getSizeOfDimension(inputGateBiasShape, 0), numUnits); in prepare()