Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.cc485 float stddev_inv = 0.0f; in PortableApplyLayerNormFloat() local
488 stddev_inv = 1.0f / sqrt(1e-8); in PortableApplyLayerNormFloat()
490 stddev_inv = 1.0f / sqrt(variance); in PortableApplyLayerNormFloat()
495 (static_cast<float>(input[index]) - mean) * stddev_inv; in PortableApplyLayerNormFloat()
734 const float stddev_inv = in PortableMeanStddevNormalization() local
737 output_vector[i] = (input_vector[i] - mean) * stddev_inv; in PortableMeanStddevNormalization()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.cc2652 const float stddev_inv = in NeonMeanStddevNormalization() local
2669 const float32x4_t output_f32x4_0 = vmulq_n_f32(tmp_0, stddev_inv); in NeonMeanStddevNormalization()
2670 const float32x4_t output_f32x4_1 = vmulq_n_f32(tmp_1, stddev_inv); in NeonMeanStddevNormalization()
2671 const float32x4_t output_f32x4_2 = vmulq_n_f32(tmp_2, stddev_inv); in NeonMeanStddevNormalization()
2672 const float32x4_t output_f32x4_3 = vmulq_n_f32(tmp_3, stddev_inv); in NeonMeanStddevNormalization()
2683 output_vector[i] = (input_vector[i] - mean) * stddev_inv; in NeonMeanStddevNormalization()