[fbsync] Import ONNX utils from symbolic_opset11 module (#4230)

Reviewed By: NicolasHug Differential Revision: D30417189 fbshipit-source-id: ca50850c51d083cd0e9e4b8c93abfba10d4af7bd
pytorch · Aug 20, 2021 · 5a24192 · 5a24192
1 parent bc72ff5
commit 5a24192
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchvision/ops/_register_onnx_ops.py b/torchvision/ops/_register_onnx_ops.py
@@ -8,7 +8,8 @@
 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):