Searched refs:batch_norm_cpu_backward_channels_last_internal (Results 1 – 1 of 1) sorted by relevance
1121 void batch_norm_cpu_backward_channels_last_internal(Tensor& grad_input, Tensor& grad_weight, Tensor… in batch_norm_cpu_backward_channels_last_internal() function1320 …batch_norm_cpu_backward_channels_last_internal<scalar_t, at::opmath_type<scalar_t>>(grad_input, gr… in batch_norm_cpu_backward_channels_last_impl()1323 …batch_norm_cpu_backward_channels_last_internal<scalar_t, scalar_t>(grad_input, grad_weight, grad_b… in batch_norm_cpu_backward_channels_last_impl()