198     const uint32_t inputRank = getNumberOfDimensions(inputShape);  in prepare()
206     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2);  in prepare()
211     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2);  in prepare()
217         NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2);  in prepare()
223     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2);  in prepare()
227     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2);  in prepare()
233         NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2);  in prepare()
239     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2);  in prepare()
243     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2);  in prepare()
257         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1);  in prepare()
263         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToForgetShape), 1);  in prepare()
269         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1);  in prepare()
287         NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1);  in prepare()
295     NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1);  in prepare()
298     NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1);  in prepare()
301     NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1);  in prepare()
306         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2);  in prepare()
313         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionBiasShape), 1);  in prepare()
318     NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2);  in prepare()
322     NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2);  in prepare()
328         NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1);  in prepare()
334         NN_RET_CHECK_EQ(getNumberOfDimensions(forgetLayerNormShape), 1);  in prepare()
340         NN_RET_CHECK_EQ(getNumberOfDimensions(cellLayerNormShape), 1);  in prepare()
346         NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1);  in prepare()