We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 87eecab commit 4248469Copy full SHA for 4248469
specdb/db.py
@@ -1282,6 +1282,21 @@
1282
OutArg(ArgType.Tensor),
1283
],
1284
),
1285
+ Spec(
1286
+ op="expm1.default", # (Tensor self) -> Tensor
1287
+ inspec=[
1288
+ InPosArg(ArgType.Tensor, name="self"),
1289
+ ],
1290
+ outspec=[
1291
+ OutArg(
1292
+ ArgType.Tensor,
1293
+ constraints=[
1294
+ cp.Dtype.In(lambda deps: dt.can_cast_from(torch.float)),
1295
+ cp.Dtype.In(lambda deps: dt.can_cast_from(deps[0].dtype)),
1296
1297
+ )
1298
1299
+ ),
1300
Spec(
1301
op="fill.Scalar", # (Tensor self, Scalar value) -> Tensor
1302
inspec=[
0 commit comments