From 0e0a372a80ac3ba94a58093c78b052e3ba28a403 Mon Sep 17 00:00:00 2001
From: Manuel Candales <mcandales@meta.com>
Date: Wed, 10 Jul 2024 09:34:56 -0700
Subject: [PATCH] SpecDB: Add OutTensor specs for convolution.default

Differential Revision: D59589875
---
 specdb/db.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/specdb/db.py b/specdb/db.py
index 5cd03d3..99f044d 100644
--- a/specdb/db.py
+++ b/specdb/db.py
@@ -1284,12 +1284,7 @@
                     cp.Value.Ge(lambda deps, length, ix: 1),
                 ],
             ),
-            InPosArg(  # transposed
-                ArgType.Bool,
-                name="transposed",
-                # TODO(mcandales): Executorch specific constraint
-                # constraints=[cp.Value.Eq(lambda deps: False)],
-            ),
+            InPosArg(ArgType.Bool, name="transposed"),
             InPosArg(  # output_padding
                 ArgType.LengthList,
                 name="output_padding",
@@ -1320,7 +1315,14 @@
             ),
         ],
         outspec=[
-            OutArg(ArgType.Tensor),
+            OutArg(
+                ArgType.Tensor,
+                constraints=[
+                    cp.Dtype.Eq(
+                        lambda deps: deps[0].dtype,
+                    ),
+                ],
+            )
         ],
     ),
     Spec(