Skip to content

Commit 1931758

Browse files
authored
fix failing test TestPyTorchConverterExamples::test_fully_dynamic_inputs (#1079)
1 parent ecf8f3d commit 1931758

File tree

1 file changed

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

1 file changed

+1
-0
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2706,6 +2706,7 @@ def copy_(context, node):
27062706
inputs = _get_inputs(context, node, expected=3)
27072707
context.add(mb.identity(x=inputs[0], name=node.name))
27082708

2709+
@register_torch_op
27092710
def dtype(context, node):
27102711
inputs = _get_inputs(context, node, expected=1)
27112712
dtype_str = inputs[0].dtype.__name__

0 commit comments

Comments
 (0)