Add Squeeze and Unsqueeze

Author: lukamac

Deeploy/OperatorDescriptor.py

@@ -440,6 +440,20 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
     attrDescriptors = [AttrDesc("axis", IntUnpack, default = 0)],
 )
 
+# Opset <= 11
+unsqueezeDesc = OperatorDescriptor(
+    inputDescriptor = IoDesc("data_in"),
+    outputDescriptor = IoDesc("data_out"),
+    attrDescriptors = [AttrDesc("axes", IntTupleUnpack)],
+)
+
+# Opset <= 11
+squeezeDesc = OperatorDescriptor(
+    inputDescriptor = IoDesc("data_in"),
+    outputDescriptor = IoDesc("data_out"),
+    attrDescriptors = [AttrDesc("axes", IntTupleUnpack)],
+)
+
 defaultOperatorDescriptors: Dict[str, OperatorDescriptor] = {
     "Add": addDesc,
     "Concat": concatDesc,
@@ -465,7 +479,9 @@ def canonicalize(self, node: gs.Node, opset: int) -> bool:
     "Slice": sliceDesc,
     "Softmax": softmaxDesc,
     "SoftmaxGrad": softmaxGradDesc,
+    "Squeeze": squeezeDesc,
     "Transpose": transposeDesc,
+    "Unsqueeze": unsqueezeDesc,
     "iHardswish": iHardswishDesc,
     "iLayerNorm": iLayerNormDesc,
     "iNoNorm": iNoNormDesc,