mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
update example in jaxpr doc
This commit is contained in:
parent
e7e4cbce5d
commit
916953ace8
@ -264,7 +264,7 @@ contains a constant ``jnp.ones(1)`` that is hoisted as a `constvar`
|
||||
let d = add c b
|
||||
in (d,) }
|
||||
linear=(False, False, False)
|
||||
true_jaxpr={ lambda ; a_ a b.
|
||||
true_jaxpr={ lambda ; e_ a b.
|
||||
let
|
||||
in (a,) } ] d e b c
|
||||
in (f,) }
|
||||
|
Loading…
x
Reference in New Issue
Block a user