We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0f82e17 commit e7f21f7Copy full SHA for e7f21f7
onnxscript/function_libs/torch_lib/ops/core.py
@@ -1651,8 +1651,8 @@ def aten_choose_qparams_optimized(
1651
def aten_chunk(self: TTensor, chunks: int, dim: int = 0) -> Sequence[TTensor]:
1652
"""chunk(Tensor(a -> *) self, int chunks, int dim=0) -> Tensor(a)[]"""
1653
if chunks == 1:
1654
- return g.op.Identity(self)
1655
- return g.op.Split(self, axis=dim, num_outputs=chunks)
+ return op.Identity(self)
+ return op.Split(self, axis=dim, num_outputs=chunks)
1656
1657
1658
@torch_op("aten::clamp", trace_only=True)
0 commit comments