Searched defs:grad_mean (Results 1 – 3 of 3) sorted by relevance
475 scalar_t grad_mean = sum / N; in batch_norm_cpu_backward_contiguous_impl() local648 Vec grad_mean = Vec::loadu(sum_data + d) / Vec(N); in batch_norm_cpu_backward_channels_last_impl() local661 Vec grad_mean = Vec::loadu(sum_data + d, n_channel - d) / Vec(N); in batch_norm_cpu_backward_channels_last_impl() local1070 opmath_t grad_mean = sum / N; in batch_norm_cpu_backward_contiguous_internal() local
448 scalar_t grad_mean = sum_a[f] / n; in batch_norm_backward_cpu_template() local
394 stat_accscalar_t grad_mean = grad_output_sum * norm; in batch_norm_backward_kernel() local