I am updating the BatchNormalizationUpdateOutputInference_kernel by adding some custom arguments for the new version.
The set of files that are updated are as follows -

./aten/src/THNN/generic/BatchNormalization.c
./aten/src/THCUNN/BatchNormalization.cu
./aten/src/THCUNN/generic/BatchNormalization.cu
./torch/legacy/nn/BatchNormalization.py
./aten/src/THNN/generic/THNN.h
./aten/src/THCUNN/generic/THCUNN.h

But stills the python setup.py build fails by
In file, ./pytorch/aten/src/ATen/nn_parse.py
RuntimeError: BatchNormalization_updateOutput: can’t find binding