RSVP for your your local TensorFlow Everywhere event today!

oryx.core.interpreters.propagate.Equation

Hashable wrapper for jax_core.Jaxprs.

params

invars Dataclass field
outvars Dataclass field
primitive Dataclass field
params_flat Dataclass field
params_tree Dataclass field

Methods

from_jaxpr_eqn

View source

__eq__