xla

950e12a6 - Added mse_loss and mse_loss_backward lowering. (#1206)

Added mse_loss and mse_loss_backward lowering. (#1206) * Added mse_loss and mse_loss_backward lowering. * Switch to map ReductionMode in tensor_methods.cpp.