Skip to content

Commit 2b1df19

Browse files
committed
Add split copy
1 parent 1c0e5cb commit 2b1df19

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
@@ -5885,7 +5885,7 @@ def _parse_keyword_args(context, node, dim, start, end, step) -> Tuple[Var]:
58855885
context.add(res)
58865886

58875887

5888-
@register_torch_op(torch_alias=["split_with_sizes", "split_with_sizes_copy"])
5888+
@register_torch_op(torch_alias=["split_with_sizes", "split_with_sizes_copy", "split_copy"])
58895889
def split(context, node):
58905890
def _parse_positional_args(context, node) -> Tuple[Var]:
58915891
inputs = _get_inputs(context, node, min_expected=2)

0 commit comments

Comments
 (0)