xla

Fix batch normalization to be able to deal with train() and eval() mo…

#806

Merged