208     const uint32_t numCells = getSizeOfDimension(inputToOutputShape, 0);  in prepare()  local
212     NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells);  in prepare()
218         NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numCells);  in prepare()
224     NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 0), numCells);  in prepare()
228     NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numCells);  in prepare()
234         NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 0), numCells);  in prepare()
240     NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 0), numCells);  in prepare()
244     NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numCells);  in prepare()
258         NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numCells);  in prepare()
264         NN_RET_CHECK_EQ(getSizeOfDimension(cellToForgetShape, 0), numCells);  in prepare()
270         NN_RET_CHECK_EQ(getSizeOfDimension(cellToOutputShape, 0), numCells);  in prepare()
288         NN_RET_CHECK_EQ(getSizeOfDimension(inputGateBiasShape, 0), numCells);  in prepare()
296     NN_RET_CHECK_EQ(getSizeOfDimension(forgetGateBiasShape, 0), numCells);  in prepare()
299     NN_RET_CHECK_EQ(getSizeOfDimension(cellGateBiasShape, 0), numCells);  in prepare()
302     NN_RET_CHECK_EQ(getSizeOfDimension(outputGateBiasShape, 0), numCells);  in prepare()
308         NN_RET_CHECK_EQ(getSizeOfDimension(projectionShape, 1), numCells);  in prepare()
324     NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 1), numCells);  in prepare()
329         NN_RET_CHECK_EQ(getSizeOfDimension(inputLayerNormShape, 0), numCells);  in prepare()
335         NN_RET_CHECK_EQ(getSizeOfDimension(forgetLayerNormShape, 0), numCells);  in prepare()
341         NN_RET_CHECK_EQ(getSizeOfDimension(cellLayerNormShape, 0), numCells);  in prepare()
347         NN_RET_CHECK_EQ(getSizeOfDimension(outputLayerNormShape, 0), numCells);  in prepare()
390         cellStateOutTensor.dimensions[1] = numCells;  in prepare()