skip_infer for skipgroupnorm

microsoft · Nov 29, 2023 · 29f1a8f · 29f1a8f
1 parent 483c490
commit 29f1a8f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnxruntime/python/tools/symbolic_shape_infer.py b/onnxruntime/python/tools/symbolic_shape_infer.py
@@ -467,6 +467,7 @@ def _onnx_infer_single_node(self, node):
             "PythonOp",
             "MultiHeadAttention",
             "GroupNorm",
+            "SkipGroupNorm",
             "BiasSplitGelu",
             "BiasAdd",
             "NhwcConv",