Searched refs:checkWeightsShape (Results 1 – 1 of 1) sorted by relevance
266 auto checkWeightsShape = [&](const RunTimeOperandInfo* weights, uint32_t columns) -> bool { in prepare() local278 NN_RET_CHECK(checkWeightsShape(inputToInputWeights, inputSize)); in prepare()279 NN_RET_CHECK(checkWeightsShape(inputToForgetWeights, inputSize)); in prepare()280 NN_RET_CHECK(checkWeightsShape(inputToCellWeights, inputSize)); in prepare()281 NN_RET_CHECK(checkWeightsShape(inputToOutputWeights, inputSize)); in prepare()287 NN_RET_CHECK(checkWeightsShape(recurrentToInputWeights, outputSize)); in prepare()288 NN_RET_CHECK(checkWeightsShape(recurrentToForgetWeights, outputSize)); in prepare()289 NN_RET_CHECK(checkWeightsShape(recurrentToCellWeights, outputSize)); in prepare()290 NN_RET_CHECK(checkWeightsShape(recurrentToOutputWeights, outputSize)); in prepare()