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
Sorry I should’ve specified. I am looking for the CPU implementation of batch_norm.