Searched refs:batch_norm_cpu_update_stats_template (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 200 std::tuple<Tensor,Tensor> batch_norm_cpu_update_stats_template( in batch_norm_cpu_update_stats_template() function
284 std::tuple<Tensor,Tensor> batch_norm_cpu_update_stats_template( in batch_norm_cpu_update_stats_template() function
299 …return batch_norm_cpu_update_stats_template<scalar_t, param_t, VarTransform>(input, running_mean, … in batch_norm_cpu_update_stats_template()
754 …return batch_norm_cpu_update_stats_template<scalar_t, opmath_t, Var>(self, running_mean, running_v… in batch_norm_update_stats_cpu()
756 …return batch_norm_cpu_update_stats_template<scalar_t, scalar_t, Var>(self, running_mean, running_v… in batch_norm_update_stats_cpu()
785 …auto save_stats = batch_norm_cpu_update_stats_template<scalar_t, opmath_t, InvStd>(self, running_m… in batch_norm_cpu_out()
795 …auto save_stats = batch_norm_cpu_update_stats_template<scalar_t, scalar_t, InvStd>(self, running_m… in batch_norm_cpu_out()
|