Searched refs:recurrentToCellShape (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceLSTM.cpp | 245 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare() local
246 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
247 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numCells); in prepare()
248 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 1), outputSize); in prepare()
|
| H A D | QLSTM.cpp | 223 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare() local
224 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
225 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numUnits); in prepare()
226 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 1), outputSize); in prepare()
|