Fix arguments

microsoft · Mar 5, 2024 · 1ee3f75 · 1ee3f75
1 parent b649022
commit 1ee3f75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -5652,11 +5652,11 @@ def aten_native_batch_norm(
     # three outputs when training_mode=True and one when it is False.
     if training is True:
         norm, mean, var = _aten_native_batch_norm_training_onnx(
-            input, weight, bias, running_mean, running_var, axes, training, momentum, eps
+            input, weight, bias, running_mean, running_var, axes, momentum=momentum, eps=eps
         )
     else:
         norm, mean, var = _aten_native_batch_norm_inference_onnx(
-            input, weight, bias, running_mean, running_var, axes, training, momentum, eps
+            input, weight, bias, running_mean, running_var, axes, momentum=momentum, eps=eps
         )
     return norm, mean, var