We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f180f85 commit 1ca2766Copy full SHA for 1ca2766
Deeploy/OperatorDescriptor.py
@@ -508,11 +508,7 @@ class SqueezeDescriptor(OperatorDescriptor):
508
509
def canonicalize(self, node: gs.Node, opset: int) -> bool:
510
if len(node.inputs) == 2:
511
- axes = node.inputs[1]
512
- assert isinstance(axes, gs.Constant), \
513
- f"Expected axes to be a constant but received axes of type {type(axes)}"
514
- node.attrs["axes"] = axes.values
515
- axes.outputs.clear()
+ inputTensorToAttr(node, node.inputs[1], "axes")
516
517
if opset >= 13 and len(node.inputs) != 2:
518
log.warning(
0 commit comments