jax.core.Jaxpr

Warning

This page was created from a pull request (#9655).

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]

pretty_print(*[, source_info, print_shapes, ...])

Attributes

constvars

invars

outvars

eqns

previous

Internal APIs

next

jax.core.ClosedJaxpr