Lines Matching refs:inputGateBuffer
657 std::vector<int16_t> inputGateBuffer(batchSize * numUnits); in execute() local
715 Sub1Vector(forgetGateBuffer.data(), batchSize * numUnits, inputGateBuffer.data()); in execute()
721 /*outputZeroPoint=*/0, inputGateBuffer.data()); in execute()
726 /*outputZeroPoint=*/0, inputGateBuffer.data()); in execute()
730 cellToInputEffectiveScaleA, cellToInputEffectiveScaleB, inputGateBuffer.data()); in execute()
733 ApplyLayerNorm(inputGateBuffer.data(), inputLayerNormBuffer, inputBiasBuffer, in execute()
735 batchSize, numUnits, inputGateBuffer.data()); in execute()
737 ApplySigmoid(inputGateBuffer.data(), batchSize, numUnits, inputGateBuffer.data()); in execute()
743 CwiseMul(inputGateBuffer.data(), cellGateBuffer.data(), batchSize, numUnits, 30 + cellShift, in execute()
774 ApplyTanh(cellShift + 15, cellStateBuffer, batchSize, numUnits, inputGateBuffer.data()); in execute()
775 CwiseMul(outputGateBuffer.data(), inputGateBuffer.data(), hiddenStateEffectiveScaleA, in execute()