I guess at::native::batch_norm_backward_reduce_kernel is used as you are reducing the loss, so the gradients are not passed as a 4D channels-last tensor.
If you use output.backward(torch.ones_like(output)) you would get:
at::native::batch_norm_backward_reduce_channels_last_kernel
at::native::batch_norm_backward_elemt_channels_last_kernel