oryx.core.interpreters.propagate.Equation

Hashable wrapper for jax_core.Jaxprs.

oryx.core.interpreters.propagate.Equation(
    invars: Tuple[jax_core.Var],
    outvars: Tuple[jax_core.Var],
    primitive: jax_core.Primitive,
    params_flat: Tuple[Any],
    params_tree: Any
)

params

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

Methods

from_jaxpr_eqn

View source

@classmethod
from_jaxpr_eqn(
    eqn
)

__eq__

__eq__(
    other
)