How to implement batch normalization in place or Activated Batch Normzalization in place?

<Refer to this for more info: https://github.com/mapillary/inplace_abn/blob/master/train_imagenet.py>