mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

[jaxpr.rst] Remove extraneous 'let' in Jaxpr grammar.

Browse Source 
Remove `let` from `Eqn` rule. `let` appears only once in the `Jaxpr` rule.

Consistently capitalize grammar nonterminals: `jaxpr` → `Jaxpr`.
This commit is contained in:
Dan Zheng 2024-02-18 09:42:33 -08:00
parent e9fdfc7204
commit e9e3c80258
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docs/jaxpr.rst
View File

@ -34,7 +34,7 @@ There are two related representations in the code for jaxprs,
:py:class:`jax.core.Jaxpr`, and is what you obtain when you use
:py:func:`jax.make_jaxpr` to inspect jaxprs. It has the following fields:
* ``jaxpr``: is a :py:class:`jax.core.Jaxpr` representing the actual
* ``jaxpr`` is a :py:class:`jax.core.Jaxpr` representing the actual
computation content of the function (described below).
* ``consts`` is a list of constants.
@ -42,9 +42,9 @@ The most interesting part of the ClosedJaxpr is the actual execution content,
represented as a :py:class:`jax.core.Jaxpr` as printed using the following
grammar::
jaxpr ::= { lambda Var* ; Var+.
let Eqn*
in [Expr+] }
Jaxpr ::= { lambda Var* ; Var+. let
Eqn*
in [Expr+] }
where:
* The parameters of the jaxpr are shown as two lists of variables separated by
@ -62,7 +62,7 @@ where:
Equations are printed as follows::
Eqn ::= let Var+ = Primitive [ Param* ] Expr+
Eqn ::= Var+ = Primitive [ Param* ] Expr+
where:
* ``Var+`` are one or more intermediate variables to be defined as the output
@ -76,7 +76,7 @@ where:
square brackets. Each parameter is shown as ``Name = Value``.
Most jaxpr primitives are first-order (they take just one or more Expr as arguments)::
Most jaxpr primitives are first-order (they take just one or more ``Expr`` as arguments)::
Primitive := add | sub | sin | mul | ...