Skip to content

Commit 0084cb1

Browse files
committed
Update core.py
1 parent 51fbe4c commit 0084cb1

File tree

1 file changed

+3
-3
lines changed
  • onnxscript/function_libs/torch_aten/ops

1 file changed

+3
-3
lines changed

onnxscript/function_libs/torch_aten/ops/core.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5740,7 +5740,7 @@ def aten_type_as(self: TensorType, other: TensorType) -> TensorType:
57405740
raise NotImplementedError()
57415741

57425742

5743-
@torch_op("aten::unfold", trace_only=True) # FIXME: seems ast.For was not supported
5743+
@torch_op("aten::unfold", trace_only=True) # FIXME: Seems ast.For was not supported
57445744
def aten_unfold(self: TTensor, dimension: int, size: int, step: int) -> TTensor:
57455745
"""unfold(Tensor(a) self, int dimension, int size, int step) -> Tensor(a)"""
57465746

@@ -5762,8 +5762,8 @@ def aten_unfold(self: TTensor, dimension: int, size: int, step: int) -> TTensor:
57625762
seq_result = op.SequenceInsert(seq_result, slice_result)
57635763
concat_result = op.ConcatFromSequence(seq_result, axis=dimension, new_axis=1)
57645764

5765-
# generate permute of the new shape
5766-
# below logic equal to:
5765+
# Generate permute of the new shape
5766+
# Below logic equal to:
57675767
# perm = [0,1,2,3,4]
57685768
# perm.append(perm.pop(dimension+1))
57695769

0 commit comments

Comments
 (0)