diff --git a/facto/specdb/db.py b/facto/specdb/db.py index 6dda077..9e0943f 100644 --- a/facto/specdb/db.py +++ b/facto/specdb/db.py @@ -5084,6 +5084,30 @@ OutArg(ArgType.Tensor), ], ), + Spec( + op="special_zeta.default", # (Tensor self, Tensor other) -> Tensor + inspec=[ + InPosArg( + ArgType.Tensor, + name="self", + constraints=[ + cp.Dtype.In(lambda deps: [torch.float, torch.double]), + ], + ), + InPosArg( + ArgType.Tensor, + name="other", + deps=[0], + constraints=[ + cp.Dtype.Eq(lambda deps: deps[0].dtype), + cp.Size.In( + lambda deps, r, d: fn.broadcast_with(deps[0].shape, r, d) + ), + ], + ), + ], + outspec=[OutArg(ArgType.Tensor)], + ), Spec( op="t_copy.default", # (Tensor self) -> Tensor inspec=[