Skip to content

Commit e7f21f7

Browse files
committed
removes g.op
1 parent 0f82e17 commit e7f21f7

File tree

1 file changed

+2
-2
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+2
-2
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1651,8 +1651,8 @@ def aten_choose_qparams_optimized(
16511651
def aten_chunk(self: TTensor, chunks: int, dim: int = 0) -> Sequence[TTensor]:
16521652
"""chunk(Tensor(a -> *) self, int chunks, int dim=0) -> Tensor(a)[]"""
16531653
if chunks == 1:
1654-
return g.op.Identity(self)
1655-
return g.op.Split(self, axis=dim, num_outputs=chunks)
1654+
return op.Identity(self)
1655+
return op.Split(self, axis=dim, num_outputs=chunks)
16561656

16571657

16581658
@torch_op("aten::clamp", trace_only=True)

0 commit comments

Comments
 (0)