Keras batch normalization fused. Which would be the correct size for the batch norm.