Update onnxruntime/core/optimizer/transpose_optimization/onnx_transpo…

…se_optimization.cc Co-authored-by: Scott McKay <[email protected]>
microsoft · Aug 17, 2023 · 532f11e · 532f11e
1 parent ff8b861
commit 532f11e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/onnxruntime/core/optimizer/transpose_optimization/onnx_transpose_optimization.cc b/onnxruntime/core/optimizer/transpose_optimization/onnx_transpose_optimization.cc
@@ -1193,7 +1193,12 @@ bool TransposeQuantizeDequantizeAxis(const api::GraphRef& graph, const std::vect
 
 constexpr bool HandleQuantizeDequantizeAxis(const api::GraphRef& graph, const std::vector<int64_t>& perm,
                                             api::NodeRef& node, int64_t opset) {
-  if (opset >= 13) {
+  if (opset < 13) {
+    // no `axis` value until opset 13
+    return true;
+  }
+
+  return TransposeQuantizeDequantizeAxis(graph, perm, node);;
     return TransposeQuantizeDequantizeAxis(graph, perm, node);
   }