Home
last modified time | relevance | path

Searched refs:variance2 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.cc433 int32_t variance2 = static_cast<int32_t>(variance / kTwoToPower20); in PortableApplyLayerNorm() local
434 if (variance2 < 1) { in PortableApplyLayerNorm()
435 variance2 = variance_limit; in PortableApplyLayerNorm()
439 GetInvSqrtQuantizedMultiplierExp(variance2, /*reverse_shift*/ -1, in PortableApplyLayerNorm()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.cc1530 int32_t variance2 = static_cast<int32>(variance / 1048576); in NeonApplyLayerNorm() local
1531 if (variance2 < 1) { in NeonApplyLayerNorm()
1532 variance2 = variance_limit; in NeonApplyLayerNorm()
1536 GetInvSqrtQuantizedMultiplierExp(variance2, /*reverse_shift*/ -1, in NeonApplyLayerNorm()