179     const uint32_t inputRank = getNumberOfDimensions(inputShape);  in prepare()
186     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2);  in prepare()
191     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2);  in prepare()
197         NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2);  in prepare()
203     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2);  in prepare()
207     NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2);  in prepare()
213         NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2);  in prepare()
219     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2);  in prepare()
223     NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2);  in prepare()
237         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1);  in prepare()
243         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToForgetShape), 1);  in prepare()
249         NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1);  in prepare()
267         NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1);  in prepare()
275     NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1);  in prepare()
278     NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1);  in prepare()
281     NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1);  in prepare()
286         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2);  in prepare()
293         NN_RET_CHECK_EQ(getNumberOfDimensions(projectionBiasShape), 1);  in prepare()
298     NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2);  in prepare()
302     NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2);  in prepare()
308         NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1);  in prepare()
314         NN_RET_CHECK_EQ(getNumberOfDimensions(forgetLayerNormShape), 1);  in prepare()
320         NN_RET_CHECK_EQ(getNumberOfDimensions(cellLayerNormShape), 1);  in prepare()
326         NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1);  in prepare()