Skip to content

Commit e7a9c5c

Browse files
axchtensorflower-gardener
authored andcommitted
Bump numerical tolerance for agreement between TF and XLA log_probs for OneHotCategorical.
PiperOrigin-RevId: 386474297
1 parent b18cf30 commit e7a9c5c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_probability/python/distributions/platform_compatibility_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@
165165
'Kumaraswamy': 4e-5,
166166
'Logistic': 3e-6,
167167
'Multinomial': 2e-4,
168-
'OneHotCategorical': 6e-6,
168+
'OneHotCategorical': 1e-5,
169169
'PowerSpherical': 2e-5,
170170
'SigmoidBeta': 5e-4,
171171
'Skellam': 1e-4,

0 commit comments

Comments
 (0)