Searched defs:inv_std (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Normalization.cuh | 1092 const accscalar_t* __restrict__ inv_std, in batch_norm_transform_input_channels_last_kernel()
1171 const accscalar_t* __restrict__ inv_std, in batch_norm_backward_reduce_channels_last_kernel()
1328 const accscalar_t* __restrict__ inv_std, in batch_norm_backward_elemt_channels_last_kernel_impl()
1382 const accscalar_t* __restrict__ inv_std, in batch_norm_backward_elemt_channels_last_kernel()
1412 const accscalar_t* __restrict__ inv_std, in batch_norm_backward_elemt_channels_last_kernel()
1472 const at::Tensor& inv_std, in batch_norm_elemt_channels_last_cuda_template()
1531 const at::Tensor& inv_std, in batch_norm_backward_reduce_cuda_channels_last_template()
1618 const at::Tensor& inv_std, in batch_norm_backward_elemt_channels_last_cuda_template()
1686 const at::Tensor& inv_std, in batch_norm_backward_elemt_channels_last_cuda_template()
|
| /aosp_15_r20/external/webrtc/common_audio/vad/ |
| H A D | vad_gmm.c | 33 int16_t tmp16, inv_std, inv_std2, exp_value = 0; in WebRtcVad_GaussianProbability() local
|
| /aosp_15_r20/external/executorch/backends/cadence/reference/operators/ |
| H A D | quantized_layer_norm.cpp | 70 float inv_std = 1.0f / std::sqrt(variance + eps); in quantized_layer_norm_per_tensor_() local
|
| /aosp_15_r20/external/executorch/backends/cadence/hifi/operators/ |
| H A D | quantized_layer_norm.cpp | 76 float inv_std = XT_RECIP_S(XT_SQRT_S(XT_ADD_S(variance, (float)eps))); in quantized_layer_norm_per_tensor_() local
|