SpecDB: Add spec: eq.Tensor

Reviewed By: SS-JIA

Differential Revision: D52924599

fbshipit-source-id: 8d3ea7776ae8a17bee65bf29b7a97f9f81d4e709

Author: manuelcandales

specdb/db.py

@@ -1263,6 +1263,23 @@
         ],
         outspec=[OutArg(ArgType.Tensor)],
     ),
+    Spec(
+        op="eq.Tensor",  # (Tensor self, Tensor other) -> Tensor
+        inspec=[
+            InPosArg(ArgType.Tensor, name="self"),
+            InPosArg(
+                ArgType.Tensor,
+                name="other",
+                deps=[0],
+                constraints=[
+                    cp.Size.In(
+                        lambda deps, r, d: fn.broadcast_with(deps[0].shape, r, d)
+                    ),
+                ],
+            ),
+        ],
+        outspec=[OutArg(ArgType.Tensor)],
+    ),
     Spec(
         op="erf.default",  # (Tensor self) -> Tensor
         inspec=[