Searched refs:forgetGateBuffer (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | QLSTM.cpp | 658 std::vector<int16_t> forgetGateBuffer(batchSize * numUnits); in execute() local
678 /*outputZeroPoint=*/0, forgetGateBuffer.data()); in execute()
683 /*outputZeroPoint=*/0, forgetGateBuffer.data()); in execute()
687 cellToForgetEffectiveScaleA, cellToForgetEffectiveScaleB, forgetGateBuffer.data()); in execute()
690 ApplyLayerNorm(forgetGateBuffer.data(), forgetLayerNormBuffer, forgetBiasBuffer, in execute()
692 numUnits, forgetGateBuffer.data()); in execute()
694 ApplySigmoid(forgetGateBuffer.data(), batchSize, numUnits, forgetGateBuffer.data()); in execute()
715 Sub1Vector(forgetGateBuffer.data(), batchSize * numUnits, inputGateBuffer.data()); in execute()
741 CwiseMul(forgetGateBuffer.data(), prevCellStateBuffer, batchSize, numUnits, in execute()
742 /*shift=*/15, forgetGateBuffer.data()); in execute()
[all …]
|