Specify axis parameter for DequantizeLinear when input rank is 1

Signed-off-by: isdanni <leedanni@gmail.com>
onnx · May 6, 2024 · 6e1e224 · 6e1e224
1 parent 9f6f3a0
commit 6e1e224
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/onnx/defs/quantization/defs.cc b/onnx/defs/quantization/defs.cc
@@ -57,9 +57,10 @@ ONNX_OPERATOR_SET_SCHEMA(
         .Output(0, "y", "N-D quantized output tensor. It has same shape as input `x`.", "T2")
         .Attr(
             "axis",
-            "(Optional) The axis of the dequantizing dimension of the input tensor. Used for per-axis and blocked "
+            "(Optional) The axis of the dequantizing dimension of the input tensor. Used only for per-axis and blocked "
             "quantization. Negative value means counting dimensions from the back. Accepted range is `[-r, r-1]` "
-            "where `r = rank(input)`.",
+            "where `r = rank(input)`. When the rank of the input is 1, per-tensor quantization is applied, "
+            "rendering the axis unnecessary in this scenario.",
             AttributeProto::INT,
             static_cast<int64_t>(1))
         .Attr(

diff --git a/onnx/defs/quantization/old.cc b/onnx/defs/quantization/old.cc
@@ -110,7 +110,10 @@ ONNX_OPERATOR_SET_SCHEMA(
         .Output(0, "y", "N-D full precision output tensor. It has same shape as input 'x'.", "T2")
         .Attr(
             "axis",
-            "(Optional) The axis of the dequantizing dimension of the input tensor. Ignored for per-tensor quantization. Negative value means counting dimensions from the back. Accepted range is [-r, r-1] where r = rank(input).",
+            "(Optional) The axis of the dequantizing dimension of the input tensor. Used only for per-axis quantization. "
+            "Negative value means counting dimensions from the back. Accepted range is `[-r, r-1]` "
+            "where `r = rank(input)`. When the rank of the input is 1, per-tensor quantization is applied, "
+            "rendering the axis unnecessary in this scenario.",
             AttributeProto::INT,
             static_cast<int64_t>(1))
         .TypeConstraint(