xla
Use the ATen batch_norm when training is not set
#235
Merged

Loading