Implementation of batch norm

I am looking for where batch norm is computed in the code on the pytorch github, but I cannot find it anywhere. Could anybody link me to the file its implemented?

@bananacode Are you looking for the kernel implementation of batch_norm, or are you looking for the Python / C++ API entry point of batch_norm?

cuDNN bindings are at

Sorry I should’ve specified. I am looking for the CPU implementation of batch_norm.

1 Like