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

Merge pull request #9343 from jakevdp:with-config

Browse Source 
PiperOrigin-RevId: 424555882
This commit is contained in:
jax authors 2022-01-27 02:23:59 -08:00
commit 6bd9292b3d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jax/test_util.py
View File

@ -27,6 +27,7 @@ from jax._src.test_util import (
format_shape_dtype_string,
rand_uniform,
skip_on_devices,
with_config,
xla_bridge,
_default_tolerance
)