Clarification of how Batch normalization works on Tensorflow

Question

To better understand how the BN works, I have decided to code my batch normalization and compare it to the TF implementation. I made a simple example (image attached). I com