onnxscript 0.7.1.dev20260522__tar.gz → 0.7.1.dev20260602__tar.gz

{onnxscript-0.7.1.dev20260522/onnxscript.egg-info → onnxscript-0.7.1.dev20260602}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.1.dev20260522
+Version: 0.7.1.dev20260602
 Summary: Naturally author ONNX functions and models using a subset of Python
 Author-email: Microsoft Corporation <onnx@microsoft.com>
 License: MIT License
@@ -27,7 +27,7 @@ License: MIT License
         
 Project-URL: Homepage, https://microsoft.github.io/onnxscript/
 Project-URL: Repository, https://github.com/microsoft/onnxscript
-Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/8fdb1e040c797068f2e829aef5444e1bf3aa992e
+Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/1fb43af420966ff10b8baabbc3c1c50514a472fc
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers

{onnxscript-0.7.1.dev20260522 → onnxscript-0.7.1.dev20260602}/onnxscript/function_libs/torch_lib/ops/core.py

@@ -614,7 +614,7 @@ def _adjust_args_for_arange_int_dtype(
 def aten_arange_start_step(
     start: TRealUnlessFloat16OrInt8,
     end: TRealUnlessFloat16OrInt8,
-    step: TRealUnlessFloat16OrInt8 = 1.0,
+    step: TRealUnlessFloat16OrInt8 = 1,
     dtype: int = -1,
     layout: str = "",
     device: str = "",

{onnxscript-0.7.1.dev20260522 → onnxscript-0.7.1.dev20260602}/onnxscript/function_libs/torch_lib/ops/nn.py

@@ -2330,6 +2330,7 @@ def _aten_upsample_output_size(
     coordinate_transformation_mode: str,
     antialias: int = 0,
     cubic_coeff_a: float = -0.75,
+    exclude_outside: int = 0,
 ) -> TReal:
     batch_and_channel = op.Shape(self, end=2, start=0)
     # When output_size is passed in as a list of integers, the torch.onnx
@@ -2348,6 +2349,7 @@ def _aten_upsample_output_size(
         cubic_coeff_a=cubic_coeff_a,
         nearest_mode="floor",
         antialias=antialias,
+        exclude_outside=exclude_outside,
     )
 
 
@@ -2358,6 +2360,7 @@ def _aten_upsample_scales(
     coordinate_transformation_mode: str,
     antialias: int = 0,
     cubic_coeff_a: float = -0.75,
+    exclude_outside: int = 0,
 ) -> TReal:
     return op.Resize(
         self,
@@ -2371,6 +2374,7 @@ def _aten_upsample_scales(
         cubic_coeff_a=cubic_coeff_a,
         nearest_mode="floor",
         antialias=antialias,
+        exclude_outside=exclude_outside,
     )
 
 
@@ -2410,6 +2414,10 @@ def aten__upsample_bicubic2d_aa(
     coordinate_transformation_mode = _get_upsample_align_corners_mode(align_corners)
     # PyTorch uses cubic_coeff_a=-0.5 (Keys interpolation, PIL-compatible) when
     # antialias=True, as opposed to -0.75 (OpenCV-compatible) for the non-antialias case.
+    # exclude_outside=1 matches PyTorch's antialias kernel, which drops out-of-bounds
+    # samples from the filter window and renormalizes the remaining weights so they
+    # sum to 1. Without it the ONNX Resize keeps phantom out-of-bounds weight in the
+    # denominator and produces values that differ from eager near the boundary.
     return _aten_upsample_output_size(
         self,
         output_size,
@@ -2417,6 +2425,7 @@ def aten__upsample_bicubic2d_aa(
         coordinate_transformation_mode=coordinate_transformation_mode,
         antialias=1,
         cubic_coeff_a=-0.5,
+        exclude_outside=1,
     )
 
 
@@ -2495,12 +2504,17 @@ def aten__upsample_bilinear2d_aa(
     # NOTE: Based on experimentation, scales_h and scales_w are always ignored in PyTorch,
     # unless when align_corners is True, in which case we do not know what is going on.
     coordinate_transformation_mode = _get_upsample_align_corners_mode(align_corners)
+    # exclude_outside=1 matches PyTorch's antialias kernel, which drops out-of-bounds
+    # samples from the filter window and renormalizes the remaining weights so they
+    # sum to 1. Without it the ONNX Resize keeps phantom out-of-bounds weight in the
+    # denominator and produces values that differ from eager near the boundary.
     return _aten_upsample_output_size(
         self,
         output_size,
         coordinate_transformation_mode=coordinate_transformation_mode,
         mode="linear",
         antialias=1,
+        exclude_outside=1,
     )
 
 

{onnxscript-0.7.1.dev20260522 → onnxscript-0.7.1.dev20260602/onnxscript.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.1.dev20260522
+Version: 0.7.1.dev20260602
 Summary: Naturally author ONNX functions and models using a subset of Python
 Author-email: Microsoft Corporation <onnx@microsoft.com>
 License: MIT License
@@ -27,7 +27,7 @@ License: MIT License
         
 Project-URL: Homepage, https://microsoft.github.io/onnxscript/
 Project-URL: Repository, https://github.com/microsoft/onnxscript
-Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/8fdb1e040c797068f2e829aef5444e1bf3aa992e
+Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/1fb43af420966ff10b8baabbc3c1c50514a472fc
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers