Skip to content

Commit f37f235

Browse files
authored
Fix up previous jaxpr.rst commit. (jax-ml#2647)
1 parent 5174f6d commit f37f235

File tree

1 file changed

+1
-13
lines changed

1 file changed

+1
-13
lines changed

docs/jaxpr.rst

+1-13
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
Understanding jaxprs
1+
Understanding Jaxprs
22
====================
33

44
Updated: February 14, 2020 (for commit 9e6fe64).
@@ -373,18 +373,6 @@ For the example consider the function ``func11`` below::
373373
return lax.scan(body, 0., (arr, ones))
374374

375375
print(api.make_jaxpr(func11)(onp.ones(16), 5.))
376-
{ lambda c ; a b.
377-
let d e = scan[ forward=True
378-
jaxpr={ lambda ; a b c d e.
379-
let f = mul c e
380-
g = add b f
381-
h = add g a
382-
in (h, b) }
383-
length=16
384-
linear=(False, False, False, True, False)
385-
num_carry=1
386-
num_consts=1 ] b 0.0 a * c
387-
in (d, e) }
388376
{ lambda c ; a b.
389377
let d e = scan[ forward=True
390378
jaxpr={ lambda ; f a b c.

0 commit comments

Comments
 (0)