Import ONNX utils from symbolic_opset11 module (#4230)

torchvision/ops/_register_onnx_ops.py

@@ -8,7 +8,8 @@ _onnx_opset_version = 11
 def _register_custom_op():
     from torch.onnx.symbolic_helper import parse_args, scalar_type_to_onnx, scalar_type_to_pytorch_type, \
         cast_pytorch_to_onnx
-    from torch.onnx.symbolic_opset9 import select, unsqueeze, squeeze, _cast_Long, reshape
+    from torch.onnx.symbolic_opset9 import _cast_Long
+    from torch.onnx.symbolic_opset11 import select, squeeze, unsqueeze
 
     @parse_args('v', 'v', 'f')
     def symbolic_multi_label_nms(g, boxes, scores, iou_threshold):