onnx 1.16.0__cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl → 1.16.1__cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

onnx/backend/test/case/node/dequantizelinear.py

@@ -202,7 +202,7 @@ class DequantizeLinear(Base):
         # scalar zero point and scale
         x = make_tensor("x", TensorProto.UINT4, [5], [0, 1, 7, 10, 15])
         x_scale = np.float32(2)
-        x_zero_point = make_tensor("zero_point", TensorProto.UINT4, (1,), [1])
+        x_zero_point = make_tensor("x_zero_point", TensorProto.UINT4, (1,), [1])
         y = np.array([-2, 0, 12, 18, 28], dtype=np.float32)
 
         expect(
@@ -224,7 +224,7 @@ class DequantizeLinear(Base):
         # scalar zero point and scale
         x = make_tensor("x", TensorProto.INT4, [5], [0, 1, 7, -4, -8])
         x_scale = np.float32(2)
-        x_zero_point = make_tensor("zero_point", TensorProto.INT4, (1,), [1])
+        x_zero_point = make_tensor("x_zero_point", TensorProto.INT4, (1,), [1])
         y = np.array([-2, 0, 12, -10, -18], dtype=np.float32)
 
         expect(

onnx/backend/test/case/node/quantizelinear.py

@@ -73,7 +73,7 @@ class QuantizeLinear(Base):
 
         x = np.array([0.0, 1.0, 2.0, 100000.0, 200.0]).astype(np.float32)
         y_scale = np.float32(2)
-        y_zero_point = make_tensor("zero_point", TensorProto.FLOAT8E4M3FN, [1], [0])
+        y_zero_point = make_tensor("y_zero_point", TensorProto.FLOAT8E4M3FN, [1], [0])
         y = make_tensor("y", TensorProto.FLOAT8E4M3FN, [5], [0, 0.5, 1, 448, 96])
 
         expect(
@@ -93,7 +93,7 @@ class QuantizeLinear(Base):
 
         x = np.array([0.0, 1.0, 2.0, 100000.0, 200.0]).astype(np.float32)
         y_scale = np.float32(2)
-        y_zero_point = make_tensor("zero_point", TensorProto.FLOAT8E5M2, [1], [0.0])
+        y_zero_point = make_tensor("y_zero_point", TensorProto.FLOAT8E5M2, [1], [0.0])
         y = make_tensor("y", TensorProto.FLOAT8E5M2, [5], [0, 0.5, 1, 49152, 96])
 
         expect(
@@ -230,7 +230,7 @@ class QuantizeLinear(Base):
 
         y_scale = np.asarray([2.0, 3.0, 4.0], dtype=np.float32)
         y_zero_point = make_tensor(
-            "zero_point", TensorProto.UINT4, y_scale.shape, np.ones_like(y_scale)
+            "y_zero_point", TensorProto.UINT4, y_scale.shape, np.ones_like(y_scale)
         )
         y = make_tensor(
             "y", TensorProto.UINT4, x.shape, [1, 2, 3, 5, -1, -1, 3, 4, 4, 5, 5, 11]
@@ -262,7 +262,7 @@ class QuantizeLinear(Base):
 
         y_scale = np.asarray([2.0, 3.0, 4.0], dtype=np.float32)
         y_zero_point = make_tensor(
-            "zero_point", TensorProto.INT4, y_scale.shape, np.ones_like(y_scale)
+            "y_zero_point", TensorProto.INT4, y_scale.shape, np.ones_like(y_scale)
         )
         y = make_tensor(
             "y", TensorProto.INT4, x.shape, [1, 2, 3, 5, -8, -6, 3, 4, 4, 5, 5, 7]

onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/input_2.pb

@@ -1,2 +1 @@
-
*

B
-zero_point
+
*

Bx_zero_point

onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/input_2.pb

@@ -1,2 +1 @@
-
*

B
-zero_point
+
*

Bx_zero_point

onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/input_2.pb

@@ -1,2 +1 @@
-*
B
-zero_point
+*
By_zero_point

onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/input_2.pb

@@ -1,2 +1 @@
-*
B
-zero_point
+*
By_zero_point

onnx/common/version.h

@@ -9,6 +9,6 @@
 namespace ONNX_NAMESPACE {
 
 // Represents the most recent release version. Updated with every release.
-constexpr const char* LAST_RELEASE_VERSION = "1.16.0";
+constexpr const char* LAST_RELEASE_VERSION = "1.16.1";
 
 } // namespace ONNX_NAMESPACE

onnx/defs/quantization/defs.cc

@@ -200,6 +200,9 @@ ONNX_OPERATOR_SET_SCHEMA(
         .SetDoc(DequantizeLinear_ver21_doc)
         .TypeAndShapeInferenceFunction([](ONNX_NAMESPACE::InferenceContext& ctx) {
           propagateElemTypeFromInputToOutput(ctx, 1, 0);
+          if (!hasInputShape(ctx, 0)) {
+            return;
+          }
           auto& input_shape = getInputShape(ctx, 0);
           updateOutputShape(ctx, 0, input_shape);
         }));

onnx/defs/quantization/old.cc

@@ -130,6 +130,9 @@ ONNX_OPERATOR_SET_SCHEMA(
         .SetDoc(DequantizeLinear_ver19_doc)
         .TypeAndShapeInferenceFunction([](ONNX_NAMESPACE::InferenceContext& ctx) {
           propagateElemTypeFromInputToOutput(ctx, 1, 0);
+          if (!hasInputShape(ctx, 0)) {
+            return;
+          }
           auto& input_shape = getInputShape(ctx, 0);
           updateOutputShape(ctx, 0, input_shape);
         }));
@@ -181,7 +184,6 @@ ONNX_OPERATOR_SET_SCHEMA(
           if (!hasInputShape(ctx, 0)) {
             return;
           }
-
           auto& input_shape = getInputShape(ctx, 0);
           updateOutputShape(ctx, 0, input_shape);
         }));

onnx/shape_inference/implementation.cc

@@ -488,29 +488,29 @@ class ShapeInferenceImplBase {
           ProcessCall(n, *(iter->second), ctx);
         } else {
           has_unsupported_op = true;
+          return;
         }
       } else {
         has_unsupported_op = true;
+        return;
       }
-      if (!has_unsupported_op) {
-        for (int i = 0; i < n.output_size(); ++i) {
-          // skip type and shape propagation for missing optional outputs.
-          if (!n.output(i).empty())
-            UpdateType(n.output(i), ctx.getOutputType(i));
-        }
-        // Constant values are tracked to improve inference/checking for subsequent nodes.
-        ProcessConstant(n);
-        // If data-propagation is enabled, partial-evaluation (aka data-propagation) is performed
-        // to improve inference/checking for subsequent nodes.
-        if (options.enable_data_propagation && schema && schema->has_data_propagation_function()) {
-          if (generated_shape_data_by_name == nullptr) {
-            fail_shape_inference(
-                "Container for generated shape data cannot be nullptr when enable_data_propagation option is set.");
-          }
-          DataPropagationContextImpl data_propagation_ctx(
-              n, value_types_by_name, input_data_by_name, *generated_shape_data_by_name);
-          schema->GetDataPropagationFunction()(data_propagation_ctx);
+      for (int i = 0; i < n.output_size(); ++i) {
+        // skip type and shape propagation for missing optional outputs.
+        if (!n.output(i).empty())
+          UpdateType(n.output(i), ctx.getOutputType(i));
+      }
+      // Constant values are tracked to improve inference/checking for subsequent nodes.
+      ProcessConstant(n);
+      // If data-propagation is enabled, partial-evaluation (aka data-propagation) is performed
+      // to improve inference/checking for subsequent nodes.
+      if (options.enable_data_propagation && schema && schema->has_data_propagation_function()) {
+        if (generated_shape_data_by_name == nullptr) {
+          fail_shape_inference(
+              "Container for generated shape data cannot be nullptr when enable_data_propagation option is set.");
         }
+        DataPropagationContextImpl data_propagation_ctx(
+            n, value_types_by_name, input_data_by_name, *generated_shape_data_by_name);
+        schema->GetDataPropagationFunction()(data_propagation_ctx);
       }
     }
     ONNX_CATCH(const ONNX_NAMESPACE::InferenceError& ex) {

onnx/version.py

@@ -1,5 +1,5 @@
 # This file is generated by setup.py. DO NOT EDIT!
 
 
-version = "1.16.0"
+version = "1.16.1"
 git_version = ""

{onnx-1.16.0.dist-info → onnx-1.16.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: onnx
-Version: 1.16.0
+Version: 1.16.1
 Summary: Open Neural Network Exchange
 Author-email: ONNX Contributors <onnx-technical-discuss@lists.lfaidata.foundation>
 License: Apache License v2.0