jax.core.Jaxpr#
- class jax.core.Jaxpr(constvars, invars, outvars, eqns, effects={}, debug_info=None)[source]#
- Parameters:
- __init__(constvars, invars, outvars, eqns, effects={}, debug_info=None)[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 atoms.eqns (
Sequence
[JaxprEqn
]) – list of equations.effects (
set
[Effect
]) – set of effects. The effects on a jaxpr are a superset of the union of the effects for each equation.debug_info (
Optional
[JaxprDebugInfo
]) – optional JaxprDebugInfo.
Methods
__init__
(constvars, invars, outvars, eqns[, ...])- type constvars:
Sequence
[Var
]
pretty_print
(*[, source_info, print_shapes, ...])- param print_effects:
replace
(*[, constvars, invars, outvars, ...])Attributes
constvars
debug_info
effects
eqns
invars
outvars