102     const uint32_t numCells = getSizeOfDimension(inputToOutputShape, 0);  in prepare()  local
106     NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells);  in prepare()
112         NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numCells);  in prepare()
118     NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 0), numCells);  in prepare()
122     NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numCells);  in prepare()
128         NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 0), numCells);  in prepare()
134     NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 0), numCells);  in prepare()
138     NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numCells);  in prepare()
152         NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numCells);  in prepare()
158         NN_RET_CHECK_EQ(getSizeOfDimension(cellToForgetShape, 0), numCells);  in prepare()
164         NN_RET_CHECK_EQ(getSizeOfDimension(cellToOutputShape, 0), numCells);  in prepare()
182         NN_RET_CHECK_EQ(getSizeOfDimension(inputGateBiasShape, 0), numCells);  in prepare()
190     NN_RET_CHECK_EQ(getSizeOfDimension(forgetGateBiasShape, 0), numCells);  in prepare()
193     NN_RET_CHECK_EQ(getSizeOfDimension(cellGateBiasShape, 0), numCells);  in prepare()
196     NN_RET_CHECK_EQ(getSizeOfDimension(outputGateBiasShape, 0), numCells);  in prepare()
202         NN_RET_CHECK_EQ(getSizeOfDimension(projectionShape, 1), numCells);  in prepare()
218     NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 1), numCells);  in prepare()
223         NN_RET_CHECK_EQ(getSizeOfDimension(inputLayerNormShape, 0), numCells);  in prepare()
229         NN_RET_CHECK_EQ(getSizeOfDimension(forgetLayerNormShape, 0), numCells);  in prepare()
235         NN_RET_CHECK_EQ(getSizeOfDimension(cellLayerNormShape, 0), numCells);  in prepare()
241         NN_RET_CHECK_EQ(getSizeOfDimension(outputLayerNormShape, 0), numCells);  in prepare()
284         cellStateOutTensor.dimensions[1] = numCells;  in prepare()