Searched refs:ApplyLayerNorm (Results 1 – 3 of 3) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | QLSTM.cpp | 690 ApplyLayerNorm(forgetGateBuffer.data(), forgetLayerNormBuffer, forgetBiasBuffer, in execute()
707 ApplyLayerNorm(cellGateBuffer.data(), cellLayerNormBuffer, cellBiasBuffer, in execute()
733 ApplyLayerNorm(inputGateBuffer.data(), inputLayerNormBuffer, inputBiasBuffer, in execute()
767 ApplyLayerNorm(outputGateBuffer.data(), outputLayerNormBuffer, outputBiasBuffer, in execute()
|
| /aosp12/packages/modules/NeuralNetworks/common/ |
| H A D | QuantUtils.h | 136 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, const int32_t* bias,
|
| H A D | QuantUtils.cpp | 10 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, const int32_t* bias, in ApplyLayerNorm() function
|