Searched refs:batch_norm_cpu_backward_contiguous_internal (Results 1 – 1 of 1) sorted by relevance
997 void batch_norm_cpu_backward_contiguous_internal(Tensor& grad_input, Tensor& grad_weight, Tensor& g… in batch_norm_cpu_backward_contiguous_internal() function1112 …batch_norm_cpu_backward_contiguous_internal<scalar_t, at::opmath_type<scalar_t>>(grad_input, grad_… in batch_norm_cpu_backward_contiguous_impl()1115 …batch_norm_cpu_backward_contiguous_internal<scalar_t, scalar_t>(grad_input, grad_weight, grad_bias, in batch_norm_cpu_backward_contiguous_impl()