From 990a60e311d74b296d9940170a8d90ef0d1898d8 Mon Sep 17 00:00:00 2001 From: Meilof Veeningen Date: Mon, 1 Jul 2024 17:28:41 +0200 Subject: [PATCH] support squeeze v21 (NB this version only adds new dtypes, see https://onnx.ai/onnx/operators/text_diff_Squeeze_13_21.html) --- onnx2torch/node_converters/squeeze.py | 1 + 1 file changed, 1 insertion(+) diff --git a/onnx2torch/node_converters/squeeze.py b/onnx2torch/node_converters/squeeze.py index 824b2913..d489a489 100644 --- 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(),