fix output type

Signed-off-by: Xavier Dupre <xadupre@microsoft.com>
onnx · Aug 7, 2023 · 8e0fdf3 · 8e0fdf3
1 parent 21aa30c
commit 8e0fdf3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docs/Changelog.md b/docs/Changelog.md
@@ -24042,7 +24042,7 @@ This version of the operator has been available since version 20 of the default
 <dl>
 <dt><tt>y</tt> : T2</dt>
 <dd>Quantized output tensor</dd>
-<dt><tt>y_scale</tt> : tensor(float)</dt>
+<dt><tt>y_scale</tt> : T1</dt>
 <dd>Output scale. It's a scalar, which means a per-tensor/layer quantization.</dd>
 <dt><tt>y_zero_point</tt> : T2</dt>
 <dd>Output zero point. It's a scalar, which means a per-tensor/layer quantization.</dd>

diff --git a/docs/Operators.md b/docs/Operators.md
@@ -7952,7 +7952,7 @@ Other versions of this operator: <a href="Changelog.md#DynamicQuantizeLinear-11"
 <dl>
 <dt><tt>y</tt> : T2</dt>
 <dd>Quantized output tensor</dd>
-<dt><tt>y_scale</tt> : tensor(float)</dt>
+<dt><tt>y_scale</tt> : T1</dt>
 <dd>Output scale. It's a scalar, which means a per-tensor/layer quantization.</dd>
 <dt><tt>y_zero_point</tt> : T2</dt>
 <dd>Output zero point. It's a scalar, which means a per-tensor/layer quantization.</dd>

diff --git a/onnx/defs/quantization/defs.cc b/onnx/defs/quantization/defs.cc
@@ -246,7 +246,7 @@ ONNX_OPERATOR_SET_SCHEMA(
             1,
             "y_scale",
             "Output scale. It's a scalar, which means a per-tensor/layer quantization.",
-            "tensor(float)")
+            "T1")
         .Output(
             2,
             "y_zero_point",