SpecDB: Add out specs for log, log10, log1p, log2

Summary: Added specs for log variants

Reviewed By: SS-JIA

Differential Revision: D52921850

fbshipit-source-id: 17b546eb7f1950e1a3a2eb811dddea83298ecabd

Author: manuelcandales

specdb/db.py

@@ -1715,7 +1715,60 @@
         inspec=[
             InPosArg(ArgType.Tensor, name="self"),
         ],
-        outspec=[OutArg(ArgType.Tensor)],
+        outspec=[
+            OutArg(
+                ArgType.Tensor,
+                constraints=[
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(torch.float)),
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(deps[0].dtype)),
+                ],
+            )
+        ],
+    ),
+    Spec(
+        op="log10.default",  # (Tensor self) -> Tensor
+        inspec=[
+            InPosArg(ArgType.Tensor, name="self"),
+        ],
+        outspec=[
+            OutArg(
+                ArgType.Tensor,
+                constraints=[
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(torch.float)),
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(deps[0].dtype)),
+                ],
+            )
+        ],
+    ),
+    Spec(
+        op="log1p.default",  # (Tensor self) -> Tensor
+        inspec=[
+            InPosArg(ArgType.Tensor, name="self"),
+        ],
+        outspec=[
+            OutArg(
+                ArgType.Tensor,
+                constraints=[
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(torch.float)),
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(deps[0].dtype)),
+                ],
+            )
+        ],
+    ),
+    Spec(
+        op="log2.default",  # (Tensor self) -> Tensor
+        inspec=[
+            InPosArg(ArgType.Tensor, name="self"),
+        ],
+        outspec=[
+            OutArg(
+                ArgType.Tensor,
+                constraints=[
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(torch.float)),
+                    cp.Dtype.In(lambda deps: dt.can_cast_from(deps[0].dtype)),
+                ],
+            )
+        ],
     ),
     Spec(
         op="logical_and.default",  # (Tensor self, Tensor other) -> Tensor