Searched refs:inputGateBuffer (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | QLSTM.cpp | 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()
[all …]
|