Lines Matching refs:weights

212                             uint8_t* weights) {  in assignWeightsSubmatrix()  argument
218 weights[(row + offset_row) * weightsDims[1] + column + offset_column] = submatrixValues[i]; in assignWeightsSubmatrix()
270 auto checkWeightsShape = [&](const RunTimeOperandInfo* weights, uint32_t columns) -> bool { in prepare() argument
271 NN_RET_CHECK_EQ(NumDimensions(weights), 2u); in prepare()
272 NN_RET_CHECK_EQ(SizeOfDimension(weights, 0), outputSize); in prepare()
273 NN_RET_CHECK_EQ(SizeOfDimension(weights, 1), columns); in prepare()
274 NN_RET_CHECK_EQ(weights->scale, weightsScale); in prepare()
275 NN_RET_CHECK_EQ(weights->zeroPoint, weightsZeroPoint); in prepare()
354 uint8_t* weights) { in concatenateWeights() argument
357 assignWeightsSubmatrix(inputToInputWeights_, 0 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
358 assignWeightsSubmatrix(inputToCellWeights_, 1 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
359 assignWeightsSubmatrix(inputToForgetWeights_, 2 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
360 assignWeightsSubmatrix(inputToOutputWeights_, 3 * outputSize, outputSize, weightsDims, weights); in concatenateWeights()
361 assignWeightsSubmatrix(recurrentToInputWeights_, 0 * outputSize, 0, weightsDims, weights); in concatenateWeights()
362 assignWeightsSubmatrix(recurrentToCellWeights_, 1 * outputSize, 0, weightsDims, weights); in concatenateWeights()
363 assignWeightsSubmatrix(recurrentToForgetWeights_, 2 * outputSize, 0, weightsDims, weights); in concatenateWeights()
364 assignWeightsSubmatrix(recurrentToOutputWeights_, 3 * outputSize, 0, weightsDims, weights); in concatenateWeights()
384 std::vector<uint8_t> weights(getNumberOfElements(weightsShape)); in eval() local
385 concatenateWeights(weightsShape.dimensions, weights.data()); in eval()
418 weights.data(), convertShapeToDims(weightsShape), bias.data(), in eval()