mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:google/jax

Browse Source
This commit is contained in:
Matthew Johnson 2019-08-02 11:26:49 -07:00
commit fd4b84bd95
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/lax_scipy_test.py
View File

@ -68,7 +68,8 @@ JAX_SPECIAL_FUNCTION_RECORDS = [
op_record("log_ndtr", 1, float_dtypes, jtu.rand_default(), True),
op_record("ndtri", 1, float_dtypes, jtu.rand_uniform(0.05, 0.95), True),
op_record("ndtr", 1, float_dtypes, jtu.rand_default(), True),
op_record("entr", 1, float_dtypes, jtu.rand_default(), True),
# TODO(phawkins): gradient of entr yields NaNs.
op_record("entr", 1, float_dtypes, jtu.rand_default(), False),
]
CombosWithReplacement = itertools.combinations_with_replacement