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 e4914b4 commit 266c94bCopy full SHA for 266c94b
backends/apple/coreml/compiler/torch_ops.py
@@ -15,6 +15,7 @@
15
_get_inputs,
16
NUM_TO_NUMPY_DTYPE,
17
NUM_TO_TORCH_DTYPE,
18
+ split,
19
transpose,
20
unbind,
21
)
@@ -37,6 +38,12 @@ def unbind_copy(context, node):
37
38
unbind(context, node)
39
40
41
+# https://github.com/apple/coremltools/pull/2563
42
+@register_torch_op(override=False)
43
+def split_copy(context, node):
44
+ split(context, node)
45
+
46
47
# https://github.com/apple/coremltools/pull/2558
48
@register_torch_op(
49
torch_alias=["torchao::dequantize_affine", "torchao.dequantize_affine"],
0 commit comments