201     const uint32_t inputRank = getNumberOfDimensions(inputShape);  in prepare()
209     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2);  in prepare()
214     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2);  in prepare()
220         NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2);  in prepare()
226     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2);  in prepare()
230     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2);  in prepare()
236         NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2);  in prepare()
242     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2);  in prepare()
246     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2);  in prepare()
260         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1);  in prepare()
266         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToForgetShape), 1);  in prepare()
272         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1);  in prepare()
290         NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1);  in prepare()
298     NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1);  in prepare()
301     NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1);  in prepare()
304     NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1);  in prepare()
309         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2);  in prepare()
316         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionBiasShape), 1);  in prepare()
321     NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2);  in prepare()
325     NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2);  in prepare()
331         NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1);  in prepare()
337         NN_RET_CHECK_EQ(getNumberOfDimensions(forgetLayerNormShape), 1);  in prepare()
343         NN_RET_CHECK_EQ(getNumberOfDimensions(cellLayerNormShape), 1);  in prepare()
349         NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1);  in prepare()