type

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

onnxscript/function_libs/torch_lib/ops/common.py

@@ -86,7 +86,7 @@ def merge_dims(dims: Sequence[int | INT64]) -> INT64:
     """Merge consecutive constant dimensions."""
 
     if not dims:
-        return op.Constant(value_ints=ir.AttrInt64("value_ints", []))
+        return op.Constant(value_ints=ir.AttrInt64s("value_ints", []))
 
     remaining_dims = list(dims)
     result_dims = []
@@ -102,7 +102,7 @@ def merge_dims(dims: Sequence[int | INT64]) -> INT64:
         else:
             # A dynamic dimension, unsqueeze and append it
             current_dim = op.Reshape(
-                current_dim, op.Constant(value_ints=ir.AttrInt64("value_ints", [1]))
+                current_dim, op.Constant(value_ints=ir.AttrInt64s("value_ints", [1]))
             )
             result_dims.append(current_dim)
     if len(result_dims) == 1: