onnxscript 0.7.1.dev20260522__tar.gz → 0.7.1.dev20260604__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.1.dev20260522
+Version: 0.7.1.dev20260604
 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/e0d3edc6eb902a2dcc07e618b055234a2ef66665
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers

{onnxscript-0.7.1.dev20260522 → onnxscript-0.7.1.dev20260604}/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 = "",
@@ -4091,7 +4091,7 @@ def aten_from_file(
 def aten_full(
     size: Union[INT64, INT32],
     fill_value: TensorType,
-    dtype: int = FLOAT.dtype,
+    dtype: int = -1,
     layout: str = "",
     device: str = "",
     pin_memory: bool = False,

{onnxscript-0.7.1.dev20260522 → onnxscript-0.7.1.dev20260604}/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.dev20260604}/onnxscript/function_libs/torch_lib/ops/vision.py

@@ -8,10 +8,12 @@
 from __future__ import annotations
 
 import warnings
-from typing import Sequence
+from typing import Optional, Sequence
 
 from onnxscript.function_libs.torch_lib.registration import torch_op
+from onnxscript.function_libs.torch_lib.tensor_typing import TFloat
 from onnxscript.onnx_opset import opset18 as op
+from onnxscript.onnx_opset import opset19
 from onnxscript.onnx_types import FLOAT, INT64
 
 _INT64_MAX = 0x7FFFFFFFFFFFFFFF
@@ -91,3 +93,36 @@ def torchvision_roi_pool(input, boxes, output_size: Sequence[int], spatial_scale
         pooled_shape=(pooled_height, pooled_width),
         spatial_scale=spatial_scale,
     )
+
+
+@torch_op("torchvision::deform_conv2d", trace_only=True)
+def torchvision_deform_conv2d(
+    input: TFloat,
+    weight: TFloat,
+    offset: TFloat,
+    mask: Optional[TFloat],
+    bias: Optional[TFloat],
+    stride_h: int,
+    stride_w: int,
+    pad_h: int,
+    pad_w: int,
+    dilation_h: int,
+    dilation_w: int,
+    groups: int,
+    offset_groups: int,
+    use_mask: bool,
+):
+    """torchvision::deform_conv2d(Tensor input, Tensor weight, Tensor offset, Tensor? mask, Tensor? bias, int stride_h, int stride_w, int pad_h, int pad_w, int dilation_h, int dilation_w, int groups, int offset_groups, bool use_mask) -> Tensor"""
+
+    return opset19.DeformConv(
+        X=input,
+        W=weight,
+        offset=offset,
+        B=bias,
+        mask=mask if use_mask else None,
+        dilations=(dilation_h, dilation_w),
+        strides=(stride_h, stride_w),
+        pads=(pad_h, pad_w, pad_h, pad_w),
+        group=groups,
+        offset_group=offset_groups,
+    )

{onnxscript-0.7.1.dev20260522 → onnxscript-0.7.1.dev20260604}/onnxscript/rewriter/rules/common/_fuse_pad_into_conv.py

@@ -146,6 +146,9 @@ class _FuseConvPadBase(orp.RewriteRuleClassBase):
         if np.any(self._pads_list[:2] + self._pads_list[x_rank : x_rank + 2]):
             self._pads_list = None
             return check_result.fail(f"{pads.name} must be zero in non-spatial dimensions.")
+        if any(p < 0 for p in self._pads_list):
+            self._pads_list = None
+            return check_result.fail(f"{pads.name} must not contain negative values.")
 
         return check_result
 

{onnxscript-0.7.1.dev20260522 → onnxscript-0.7.1.dev20260604}/onnxscript/rewriter/rules/common/_fuse_relus_clips.py

@@ -74,7 +74,9 @@ class _FuseReluClipBase(RewriteRuleClassBase, abc.ABC):
                 min_clip = min_input.const_value.numpy()
 
         if len(node.inputs) > 2:
-            max_clip = node.inputs[2].const_value.numpy()
+            max_clip = node.inputs[2]
+            if max_clip is not None:
+                max_clip = max_clip.const_value.numpy()
 
         return min_clip, max_clip, dtype
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.1.dev20260522
+Version: 0.7.1.dev20260604
 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/e0d3edc6eb902a2dcc07e618b055234a2ef66665
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers