Lines Matching refs:getNumberOfDimensions
64 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
71 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2u); in prepare()
76 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2u); in prepare()
82 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2u); in prepare()
88 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2u); in prepare()
92 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2u); in prepare()
98 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2u); in prepare()
104 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2u); in prepare()
108 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2u); in prepare()
122 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1u); in prepare()
128 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToForgetShape), 1u); in prepare()
134 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1u); in prepare()
152 NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1u); in prepare()
160 NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1u); in prepare()
163 NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1u); in prepare()
166 NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1u); in prepare()
171 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2u); in prepare()
178 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionBiasShape), 1u); in prepare()
183 NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2u); in prepare()
187 NN_RET_CHECK_EQ(getNumberOfDimensions(cellStateShape), 2u); in prepare()
193 NN_RET_CHECK_EQ(getNumberOfDimensions(inputLayerNormShape), 1u); in prepare()
199 NN_RET_CHECK_EQ(getNumberOfDimensions(forgetLayerNormShape), 1u); in prepare()
205 NN_RET_CHECK_EQ(getNumberOfDimensions(cellLayerNormShape), 1u); in prepare()
211 NN_RET_CHECK_EQ(getNumberOfDimensions(outputLayerNormShape), 1u); in prepare()