jax.extend.core.JaxprEqn

class jax.extend.core.JaxprEqn(invars, outvars, primitive, params, effects, source_info, ctx)

Parameters:

• invars (list[Atom])

• outvars (list[Var])

• primitive (Primitive)

• params (dict[str, Any])

• effects (Effects)

• source_info (source_info_util.SourceInfo)

• ctx (JaxprEqnContext)

__init__(invars, outvars, primitive, params, effects, source_info, ctx)

Methods

__init__(invars, outvars, primitive, params, ...)
replace([invars, outvars, primitive, ...])

Attributes

invars
outvars
primitive
params
effects
source_info
ctx