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
)

Attributes
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
)