Searched refs:batch_norm_cpu_backward_kernel (Results 1 – 1 of 1) sorted by relevance
1372 void batch_norm_cpu_backward_kernel(Tensor& grad_input, Tensor& grad_weight, Tensor& grad_bias, in batch_norm_cpu_backward_kernel() function1401 REGISTER_DISPATCH(batch_norm_cpu_backward_stub, &batch_norm_cpu_backward_kernel);