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 ecf8f3d commit 1931758Copy full SHA for 1931758
coremltools/converters/mil/frontend/torch/ops.py
@@ -2706,6 +2706,7 @@ def copy_(context, node):
2706
inputs = _get_inputs(context, node, expected=3)
2707
context.add(mb.identity(x=inputs[0], name=node.name))
2708
2709
+@register_torch_op
2710
def dtype(context, node):
2711
inputs = _get_inputs(context, node, expected=1)
2712
dtype_str = inputs[0].dtype.__name__
0 commit comments