Searched defs:norm_fct (Results 1 – 2 of 2) sorted by relevance
517 const stat_accscalar_t norm_fct) { in batch_norm_backward_elemt_kernel_impl()561 const stat_accscalar_t norm_fct = in batch_norm_backward_elemt_kernel() local577 const stat_accscalar_t norm_fct) { in batch_norm_backward_elemt_kernel()879 auto norm_fct = static_cast<stat_accscalar_t>(1.0 / reduction_size); in batch_norm_backward_elemt_cuda_template() local1333 const accscalar_t norm_fct, in batch_norm_backward_elemt_channels_last_kernel_impl()1397 auto norm_fct = static_cast<accscalar_t>(1) / static_cast<accscalar_t>(total_numel); in batch_norm_backward_elemt_channels_last_kernel() local1417 const accscalar_t norm_fct, in batch_norm_backward_elemt_channels_last_kernel()1692 auto norm_fct = 1.0 / reduction_size; in batch_norm_backward_elemt_channels_last_cuda_template() local
239 auto norm_fct = static_cast<accscalar_t>(1.0 / (input.numel() /input.size(1)) ); in batch_norm_elementwise_backward_train() local