Searched refs:checkWeightsShape (Results 1 – 1 of 1) sorted by relevance
269 auto checkWeightsShape = [&](const RunTimeOperandInfo* weights, uint32_t columns) -> bool { in prepare() local281 NN_RET_CHECK(checkWeightsShape(inputToInputWeights, inputSize)); in prepare()282 NN_RET_CHECK(checkWeightsShape(inputToForgetWeights, inputSize)); in prepare()283 NN_RET_CHECK(checkWeightsShape(inputToCellWeights, inputSize)); in prepare()284 NN_RET_CHECK(checkWeightsShape(inputToOutputWeights, inputSize)); in prepare()290 NN_RET_CHECK(checkWeightsShape(recurrentToInputWeights, outputSize)); in prepare()291 NN_RET_CHECK(checkWeightsShape(recurrentToForgetWeights, outputSize)); in prepare()292 NN_RET_CHECK(checkWeightsShape(recurrentToCellWeights, outputSize)); in prepare()293 NN_RET_CHECK(checkWeightsShape(recurrentToOutputWeights, outputSize)); in prepare()