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 pull request #12496 from mattjj:improve-leak-checker-2

Browse Source 
PiperOrigin-RevId: 476907407
This commit is contained in:
jax authors 2022-09-26 08:50:13 -07:00
commit 28672cca0e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/nn_test.py
View File

@ -227,7 +227,7 @@ class NNFunctionsTest(jtu.JaxTestCase):
a = jnp.array(1., 'float32')
def f(hx, _):
hx = jax.nn.relu(hx + a)
hx = sigmoid(hx + a)
return hx, None
hx = jnp.array(0., 'float32')