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

fix typo in test

Browse Source
This commit is contained in:
Matthew Johnson 2022-09-23 12:42:15 -07:00
parent e76aa77895
commit 03abcc7c5c
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')