Lines Matching refs:getNumberOfDimensions
91 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
100 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2u); in prepare()
105 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2u); in prepare()
111 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2u); in prepare()
117 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2u); in prepare()
121 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2u); in prepare()
127 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2u); in prepare()
133 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2u); in prepare()
137 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2u); in prepare()
151 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1u); in prepare()
157 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToForgetShape), 1u); in prepare()
163 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1u); in prepare()
181 NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1u); in prepare()
189 NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1u); in prepare()
192 NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1u); in prepare()
195 NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1u); in prepare()
200 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2u); in prepare()
207 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionBiasShape), 1u); in prepare()
212 NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2u); in prepare()
216 NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2u); in prepare()
222 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1u); in prepare()
228 NN_RET_CHECK_EQ(getNumberOfDimensions(forgetLayerNormShape), 1u); in prepare()
234 NN_RET_CHECK_EQ(getNumberOfDimensions(cellLayerNormShape), 1u); in prepare()
240 NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1u); in prepare()