Skip to content

Commit a5cad9f

Browse files
committed
lint and comments
1 parent cab6af6 commit a5cad9f

File tree

1 file changed

+4
-0
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+4
-0
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5432,6 +5432,10 @@ def aten_narrow_copy(self: TensorType, dim: int, start: INT64, length: INT64) ->
54325432
raise NotImplementedError()
54335433

54345434

5435+
# NOTE: https://github.com/pytorch/pytorch/blob/a44f8894fa6d973693aab44a3dda079a168b05c1/torch/_decomp/decompositions.py#L1501-L1510
5436+
# _native_batch_norm_legit_no_training and _native_batch_norm_legit are meant to
5437+
# replace native_batch_norm within unknown time period.
5438+
# TODO: Refactor this after native_batch_norm is deprecated.
54355439
@torch_op("aten::_native_batch_norm_legit_no_training", trace_only=True)
54365440
def aten_native_batch_norm_no_training(
54375441
input: TFloat,

0 commit comments

Comments
 (0)