Skip to content

Commit 7187619

Browse files
manuelcandalesfacebook-github-bot
authored andcommitted
Calibrate clamp.Tensor (#32)
Summary: Pull Request resolved: #32 imported-using-ghimport Test Plan: Imported from OSS Rollback Plan: Reviewed By: digantdesai Differential Revision: D80466683 Pulled By: manuelcandales fbshipit-source-id: 2877dc61b83c28767198175d09fd0d7a978e5676
1 parent 0311e76 commit 7187619

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

facto/specdb/db.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1120,6 +1120,14 @@
11201120
deps=[0, 1],
11211121
constraints=[
11221122
cp.Optional.Eq(lambda deps: False if deps[1] is None else True),
1123+
cp.Dtype.Ne(
1124+
lambda deps: (
1125+
torch.bool
1126+
if deps[0].dtype == torch.bool
1127+
and (deps[1] is None or deps[1].dtype == torch.bool)
1128+
else None
1129+
)
1130+
),
11231131
cp.Size.In(
11241132
lambda deps, r, d: fn.broadcast_with(
11251133
(

0 commit comments

Comments
 (0)