Lines Matching refs:numUnits
188 const uint32_t numUnits = getSizeOfDimension(inputToOutputShape, 0); in prepare() local
192 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numUnits); in prepare()
198 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numUnits); in prepare()
204 NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 0), numUnits); in prepare()
208 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numUnits); in prepare()
214 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 0), numUnits); in prepare()
220 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 0), numUnits); in prepare()
224 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numUnits); in prepare()
238 NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numUnits); in prepare()
244 NN_RET_CHECK_EQ(getSizeOfDimension(cellToForgetShape, 0), numUnits); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(cellToOutputShape, 0), numUnits); in prepare()
268 NN_RET_CHECK_EQ(getSizeOfDimension(inputGateBiasShape, 0), numUnits); in prepare()
276 NN_RET_CHECK_EQ(getSizeOfDimension(forgetGateBiasShape, 0), numUnits); in prepare()
279 NN_RET_CHECK_EQ(getSizeOfDimension(cellGateBiasShape, 0), numUnits); in prepare()
282 NN_RET_CHECK_EQ(getSizeOfDimension(outputGateBiasShape, 0), numUnits); in prepare()
288 NN_RET_CHECK_EQ(getSizeOfDimension(projectionShape, 1), numUnits); in prepare()
304 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 1), numUnits); in prepare()
309 NN_RET_CHECK_EQ(getSizeOfDimension(inputLayerNormShape, 0), numUnits); in prepare()
315 NN_RET_CHECK_EQ(getSizeOfDimension(forgetLayerNormShape, 0), numUnits); in prepare()
321 NN_RET_CHECK_EQ(getSizeOfDimension(cellLayerNormShape, 0), numUnits); in prepare()
327 NN_RET_CHECK_EQ(getSizeOfDimension(outputLayerNormShape, 0), numUnits); in prepare()
392 const uint32_t numUnits = inputToOutputWeightsShape.dimensions[0]; in execute() local
655 std::vector<int16_t> inputGateBuffer(batchSize * numUnits); in execute()
656 std::vector<int16_t> forgetGateBuffer(batchSize * numUnits); in execute()
657 std::vector<int16_t> cellGateBuffer(batchSize * numUnits); in execute()
658 std::vector<int16_t> outputGateBuffer(batchSize * numUnits); in execute()
659 std::vector<int8_t> buffer8(batchSize * numUnits); in execute()
675 numUnits, in execute()
680 outputSize, numUnits, in execute()
690 numUnits, forgetGateBuffer.data()); in execute()
692 ApplySigmoid(forgetGateBuffer.data(), batchSize, numUnits, forgetGateBuffer.data()); in execute()
697 inputToCellEffectiveScaleB, batchSize, inputSize, numUnits, in execute()
702 numUnits, in execute()
707 numUnits, cellGateBuffer.data()); in execute()
709 ApplyTanh<3>(cellGateBuffer.data(), batchSize, numUnits, cellGateBuffer.data()); in execute()
713 Sub1Vector(forgetGateBuffer.data(), batchSize * numUnits, inputGateBuffer.data()); in execute()
718 numUnits, in execute()
723 recurrentToInputEffectiveScaleB, batchSize, outputSize, numUnits, in execute()
733 batchSize, numUnits, inputGateBuffer.data()); in execute()
735 ApplySigmoid(inputGateBuffer.data(), batchSize, numUnits, inputGateBuffer.data()); in execute()
739 CwiseMul(forgetGateBuffer.data(), prevCellStateBuffer, batchSize, numUnits, in execute()
741 CwiseMul(inputGateBuffer.data(), cellGateBuffer.data(), batchSize, numUnits, 30 + cellShift, in execute()
743 CwiseAdd(forgetGateBuffer.data(), cellGateBuffer.data(), batchSize, numUnits, cellStateBuffer); in execute()
745 CwiseClipping(cellStateBuffer, quantizedCellClip, batchSize, numUnits); in execute()
752 numUnits, in execute()
757 outputSize, numUnits, in execute()
767 numUnits, outputGateBuffer.data()); in execute()
769 ApplySigmoid(outputGateBuffer.data(), batchSize, numUnits, outputGateBuffer.data()); in execute()
772 ApplyTanh(cellShift + 15, cellStateBuffer, batchSize, numUnits, inputGateBuffer.data()); in execute()
774 hiddenStateEffectiveScaleB, batchSize, numUnits, hiddenStateZeroPoint, buffer8.data()); in execute()
781 projectionEffectiveScaleB, batchSize, numUnits, in execute()