Skip to content

Commit 35edb4d

Browse files
committed
Add split copy
1 parent 3d6fd01 commit 35edb4d

File tree

1 file changed

+1
-1
lines changed
  • coremltools/converters/mil/frontend/torch

1 file changed

+1
-1
lines changed

coremltools/converters/mil/frontend/torch/ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5896,7 +5896,7 @@ def _parse_keyword_args(context, node, dim, start, end, step) -> Tuple[Var]:
58965896
context.add(res)
58975897

58985898

5899-
@register_torch_op(torch_alias=["split_with_sizes", "split_with_sizes_copy"])
5899+
@register_torch_op(torch_alias=["split_with_sizes", "split_with_sizes_copy", "split_copy"])
59005900
def split(context, node):
59015901
def _parse_positional_args(context, node) -> Tuple[Var]:
59025902
inputs = _get_inputs(context, node, min_expected=2)

0 commit comments

Comments
 (0)