Home
last modified time | relevance | path

Searched defs:_var_sum (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dbatch_norm_kernel.cpp207 accscalar_t _var_sum = 0; in batch_norm_cpu_collect_stats_contiguous_impl() local
300 accscalar_t _var_sum = 0; in batch_norm_cpu_collect_stats_channels_last_impl() local
392 accscalar_t _var_sum = 0; in batch_norm_cpu_collect_stats_channels_last_impl() local
976 opmath_t _var_sum = 0; in batch_norm_cpu_collect_stats_channels_last_internal() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNormalization.cpp222 auto _var_sum = at::empty({n_input}, input.options().dtype(dtype)); in batch_norm_cpu_update_stats_template() local