Searched refs:inputToCellShape (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceLSTM.cpp | 229 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare() local
230 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
231 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numCells); in prepare()
232 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 1), inputSize); in prepare()
|
| H A D | QLSTM.cpp | 207 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare() local
208 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
209 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numUnits); in prepare()
210 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 1), inputSize); in prepare()
|