180     const uint32_t inputRank = getNumberOfDimensions(inputShape);  in prepare()
187     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2);  in prepare()
192     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2);  in prepare()
198         NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2);  in prepare()
204     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2);  in prepare()
208     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2);  in prepare()
214         NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2);  in prepare()
220     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2);  in prepare()
224     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2);  in prepare()
238         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1);  in prepare()
244         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToForgetShape), 1);  in prepare()
250         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1);  in prepare()
268         NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1);  in prepare()
276     NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1);  in prepare()
279     NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1);  in prepare()
282     NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1);  in prepare()
287         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2);  in prepare()
294         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionBiasShape), 1);  in prepare()
299     NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2);  in prepare()
303     NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2);  in prepare()
309         NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1);  in prepare()
315         NN_RET_CHECK_EQ(getNumberOfDimensions(forgetLayerNormShape), 1);  in prepare()
321         NN_RET_CHECK_EQ(getNumberOfDimensions(cellLayerNormShape), 1);  in prepare()
327         NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1);  in prepare()