jax.core.Jaxpr¶
-
class
jax.core.
Jaxpr
(constvars, invars, outvars, eqns)[source]¶ - Parameters
-
__init__
(constvars, invars, outvars, eqns)[source]¶ - Parameters
constvars (
Sequence
[Var
]) – list of variables introduced for constants. Array constants are replaced with such variables while scalar constants are kept inline.invars (
Sequence
[Var
]) – list of input variables. Together, constvars and invars are the inputs to the Jaxpr.outvars (
Sequence
[Union
[Var
,Literal
]]) – list of output variables.eqns (
Sequence
[JaxprEqn
]) – list of equations.
Methods
__init__
(constvars, invars, outvars, eqns)- type constvars
Sequence
[Var
]