support squeeze v21

(NB this version only adds new dtypes, see https://onnx.ai/onnx/operators/text_diff_Squeeze_13_21.html)
ENOT-AutoDL · Jul 1, 2024 · 990a60e · 990a60e
1 parent cd98062
commit 990a60e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnx2torch/node_converters/squeeze.py b/onnx2torch/node_converters/squeeze.py
@@ -93,6 +93,7 @@ def _(node: OnnxNode, graph: OnnxGraph) -> OperationConverterResult:  # pylint:
 
 
 @add_converter(operation_type='Squeeze', version=13)
+@add_converter(operation_type='Squeeze', version=21)
 def _(node: OnnxNode, graph: OnnxGraph) -> OperationConverterResult:  # pylint: disable=unused-argument
     return OperationConverterResult(
         torch_module=OnnxSqueezeDynamicAxes(),