fix failing test TestPyTorchConverterExamples::test_fully_dynamic_inp…

…uts (#1079)
apple · Feb 5, 2021 · 1931758 · 1931758
1 parent ecf8f3d
commit 1931758
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/coremltools/converters/mil/frontend/torch/ops.py b/coremltools/converters/mil/frontend/torch/ops.py
@@ -2706,6 +2706,7 @@ def copy_(context, node):
     inputs = _get_inputs(context, node, expected=3)
     context.add(mb.identity(x=inputs[0], name=node.name))
 
+@register_torch_op
 def dtype(context, node):
     inputs = _get_inputs(context, node, expected=1)
     dtype_str = inputs[0].dtype.__name__