Fix Op(_native_batch_norm_legit_no_training) | feat(torchlib) (#1126)

From the review comment: #1116 (comment).
microsoft · Nov 2, 2023 · aab5517 · aab5517
1 parent 132fab1
commit aab5517
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -5423,7 +5423,7 @@ def aten_narrow_copy(self: TensorType, dim: int, start: INT64, length: INT64) ->
 # replace native_batch_norm within unknown time period.
 # TODO: Refactor this after native_batch_norm is deprecated.
 @torch_op("aten::_native_batch_norm_legit_no_training", trace_only=True)
-def aten_native_batch_norm_no_training(
+def aten__native_batch_norm_no_training(
     input: TFloat,
     weight: Optional[TFloat] = None,
     bias: Optional[TFloat] = None,