oryx.core.interpreters.propagate.Environment

Keeps track of variables and their values during propagation.

Inherits From: Pytree

oryx.core.interpreters.propagate.Environment(
    cell_type, jaxpr
)

Methods

assert_same_type(
    other_env
) -> None

Raises an error if environments do not have matching Jaxprs.

copy() -> "Environment"

flatten()

join(
    other: "Environment"
) -> "Environment"

read(
    var: VarOrLiteral
) -> oryx.core.interpreters.propagate.Cell

@classmethod
unflatten(
    data, xs
)

write(
    var: VarOrLiteral,
    cell: oryx.core.interpreters.propagate.Cell
) -> oryx.core.interpreters.propagate.Cell

write_subenv(
    eqn: oryx.core.interpreters.propagate.Equation,
    subenv: "Environment"
) -> None

__contains__(
    var: VarOrLiteral
)

__getitem__(
    var: VarOrLiteral
) -> oryx.core.interpreters.propagate.Cell