Searched refs:ApplyLayerNorm (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QLSTM.cpp | 574 ApplyLayerNorm(forgetGateBuffer.data(), forgetLayerNormBuffer, forgetBiasBuffer, in execute()
591 ApplyLayerNorm(cellGateBuffer.data(), cellLayerNormBuffer, cellBiasBuffer, in execute()
617 ApplyLayerNorm(inputGateBuffer.data(), inputLayerNormBuffer, inputBiasBuffer, in execute()
651 ApplyLayerNorm(outputGateBuffer.data(), outputLayerNormBuffer, outputBiasBuffer, in execute()
|
| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.h | 140 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, const int32_t* bias,
|
| D | QuantUtils.cpp | 10 void ApplyLayerNorm(const int16_t* input, const int16_t* layer_norm_weights, const int32_t* bias, in ApplyLayerNorm() function
|